shared config
This commit is contained in:
@@ -1,18 +1,14 @@
|
||||
import os
|
||||
|
||||
import assets
|
||||
import sensors
|
||||
from config import APP
|
||||
from dagster_polars import PolarsParquetIOManager
|
||||
from icecream import install
|
||||
from resources import JsonIOManager
|
||||
from shared.config import APP, STORAGE_DIR
|
||||
|
||||
import dagster as dg
|
||||
|
||||
install()
|
||||
|
||||
storage_dir = os.environ.get("STORAGE_DIR", "/storage")
|
||||
|
||||
definitions = dg.Definitions(
|
||||
assets=[
|
||||
asset.with_attributes(
|
||||
@@ -22,8 +18,8 @@ definitions = dg.Definitions(
|
||||
for asset in dg.load_assets_from_modules([assets])
|
||||
],
|
||||
resources={
|
||||
"json_io_manager": JsonIOManager(base_dir=storage_dir),
|
||||
"polars_parquet_io_manager": PolarsParquetIOManager(base_dir=storage_dir),
|
||||
"json_io_manager": JsonIOManager(base_dir=STORAGE_DIR),
|
||||
"polars_parquet_io_manager": PolarsParquetIOManager(base_dir=STORAGE_DIR),
|
||||
},
|
||||
sensors=[sensors.list_locations, sensors.list_latitudes, sensors.retrieve_weather],
|
||||
schedules=[
|
||||
|
||||
Reference in New Issue
Block a user