Restore advanced Dagster implementation with partitioned assets