fix paths to storage and logs
This commit is contained in:
@@ -1,4 +1,3 @@
|
||||
import pandas as pd
|
||||
import polars as pl
|
||||
|
||||
from dagster import (
|
||||
@@ -23,16 +22,6 @@ partitions_def_multi = MultiPartitionsDefinition(
|
||||
)
|
||||
|
||||
|
||||
@asset(
|
||||
# tags={
|
||||
# "dagster/executor": "other_executor"
|
||||
# },
|
||||
)
|
||||
def dummy_asset():
|
||||
"""A dummy asset to ensure the module is recognized by Dagster."""
|
||||
return pd.DataFrame({"dummy": [1, 2, 3]})
|
||||
|
||||
|
||||
@asset(
|
||||
io_manager_key="polars_parquet_io_manager",
|
||||
partitions_def=partitions_def_single,
|
||||
@@ -58,7 +47,6 @@ def asset_single_1(context):
|
||||
def asset_multi_1(context):
|
||||
ic()
|
||||
ic(context.partition_key)
|
||||
|
||||
return pl.DataFrame(
|
||||
[{"date": context.partition_key, "data": f"Data for {context.partition_key}"}]
|
||||
)
|
||||
|
||||
@@ -9,7 +9,7 @@ install()
|
||||
# Define a job that includes both assets
|
||||
daily_job = define_asset_job(
|
||||
"daily_job",
|
||||
selection=[assets.dummy_asset, assets.asset_multi_1, assets.asset_multi_2],
|
||||
selection=[assets.asset_multi_1, assets.asset_multi_2],
|
||||
)
|
||||
|
||||
definitions = Definitions(
|
||||
@@ -20,6 +20,8 @@ definitions = Definitions(
|
||||
)
|
||||
for asset in load_assets_from_modules([assets])
|
||||
],
|
||||
resources={"polars_parquet_io_manager": PolarsParquetIOManager()},
|
||||
resources={
|
||||
"polars_parquet_io_manager": PolarsParquetIOManager(base_dir="/storage")
|
||||
},
|
||||
jobs=[daily_job],
|
||||
)
|
||||
|
||||
@@ -11,7 +11,6 @@ from dagster._core.instance import DynamicPartitionsStore
|
||||
from dagster._serdes import whitelist_for_serdes
|
||||
|
||||
|
||||
# @whitelist_for_serdes
|
||||
class LatestTwoPartitionsMapping(PartitionMapping):
|
||||
def get_upstream_mapped_partitions_result_for_partitions(
|
||||
self,
|
||||
|
||||
@@ -110,7 +110,7 @@ def deals(context):
|
||||
@asset(deps=[deals], io_manager_key="polars_parquet_io_manager")
|
||||
def new_deals(context: OpExecutionContext) -> pl.DataFrame:
|
||||
ic()
|
||||
storage_dir = context.instance.storage_directory()
|
||||
storage_dir = context.resources.polars_parquet_io_manager.base_dir
|
||||
asset_key = "deals"
|
||||
|
||||
with duckdb.connect() as con:
|
||||
|
||||
@@ -29,7 +29,7 @@ definitions = Definitions(
|
||||
for asset in load_assets_from_modules([assets])
|
||||
],
|
||||
resources={
|
||||
"polars_parquet_io_manager": PolarsParquetIOManager(),
|
||||
"polars_parquet_io_manager": PolarsParquetIOManager(base_dir="/storage"),
|
||||
"duckdb_io_manager": PandasDuckDBIOManager(database="vinyl"),
|
||||
},
|
||||
jobs=[deals_job, check_partititions_job, musicbrainz_lookup_job],
|
||||
|
||||
@@ -25,6 +25,7 @@ services:
|
||||
DAGSTER_CURRENT_IMAGE: user_code_vinyl
|
||||
volumes:
|
||||
- /opt/dagster/apps/:/apps/:ro
|
||||
- /opt/dagster/logs/:/logs:rw
|
||||
- /opt/dagster/storage/import/:/storage/import/:ro
|
||||
- /opt/dagster/storage/deals/:/storage/deals/:rw
|
||||
networks:
|
||||
@@ -45,5 +46,6 @@ services:
|
||||
DAGSTER_CURRENT_IMAGE: user_code_other
|
||||
volumes:
|
||||
- /opt/dagster/apps/:/apps:ro
|
||||
- /opt/dagster/logs/:/logs:rw
|
||||
networks:
|
||||
- dagster
|
||||
|
||||
@@ -19,6 +19,7 @@ x-volumes: &volumes
|
||||
- /opt/dagster/dagster.yaml:/opt/dagster/home/dagster.yaml:ro
|
||||
- /opt/dagster/workspace.yaml:/opt/dagster/home/workspace.yaml:ro
|
||||
- /opt/dagster/storage/:/storage/:rw
|
||||
- /opt/dagster/logs/:/logs:rw
|
||||
- /var/run/docker.sock:/var/run/docker.sock:rw
|
||||
|
||||
services:
|
||||
|
||||
@@ -23,6 +23,7 @@ run_launcher:
|
||||
volumes:
|
||||
- /opt/dagster/apps/:/apps:ro
|
||||
- /opt/dagster/storage/:/storage/:rw
|
||||
- /opt/dagster/logs/:/logs:rw
|
||||
|
||||
run_storage:
|
||||
module: dagster_postgres.run_storage
|
||||
@@ -71,3 +72,9 @@ event_log_storage:
|
||||
env: DAGSTER_POSTGRES_PASSWORD
|
||||
db_name:
|
||||
env: DAGSTER_POSTGRES_DB
|
||||
|
||||
compute_logs:
|
||||
module: dagster.core.storage.local_compute_log_manager
|
||||
class: LocalComputeLogManager
|
||||
config:
|
||||
base_dir: /logs
|
||||
|
||||
Reference in New Issue
Block a user