import collections
import dagstermill as dm
from dagster import pipeline, execute_pipeline, lambda_solid
from dagster import PipelineDefinition,InputDefinition, OutputDefinition, Int
from dagster import Any, Field, lambda_solid, solid, pipeline, as_dagster_type
from dagster import InputDefinition, OutputDefinition, Int
import dagstermill as dm

my_notebook_solid = dm.define_dagstermill_solid(
                            name='DM1',
                            notebook_path='DM1.ipynb',
                            input_defs = [
                                InputDefinition(name='a'),
                                InputDefinition(name='b')
                            ],
                            output_defs  = [OutputDefinition()]
                        )
#def notebook_pipeline():
   # return PipelineDefinition(name='julia_pipeline', solid_defs=[my_notebook_solid])
def notebook_pipeline():
    return PipelineDefinition(name='pipeline', solid_defs=[my_notebook_solid])