@@ -35,9 +35,9 @@ PYBIND11_MODULE(aika, m)
3535
3636 // Bind FieldDefinition first
3737 py::class_<FieldDefinition>(m, " FieldDefinition" )
38- .def (" in " , &FieldDefinition::in , py::return_value_policy::reference_internal,
38+ .def (" input " , &FieldDefinition::input , py::return_value_policy::reference_internal,
3939 py::arg (" relation" ), py::arg (" input" ), py::arg (" arg" ))
40- .def (" out " , &FieldDefinition::out , py::return_value_policy::reference_internal,
40+ .def (" output " , &FieldDefinition::output , py::return_value_policy::reference_internal,
4141 py::arg (" relation" ), py::arg (" output" ), py::arg (" arg" ));
4242
4343 // Bind AbstractFunctionDefinition (inherits from FieldDefinition)
@@ -46,7 +46,7 @@ PYBIND11_MODULE(aika, m)
4646 // Bind Subtraction (inherits from AbstractFunctionDefinition)
4747 py::class_<Subtraction, AbstractFunctionDefinition>(m, " Subtraction" );
4848
49- py::class_<InputField>(m, " InputField" )
49+ py::class_<InputField, FieldDefinition >(m, " InputField" )
5050 .def (py::init<Type*, const std::string &>())
5151 .def (" __str__" , [](const InputField &f) {
5252 return f.toString ();
@@ -76,5 +76,6 @@ PYBIND11_MODULE(aika, m)
7676 py::class_<TypeRegistry>(m, " TypeRegistry" )
7777 .def (py::init<>())
7878 .def (" getType" , &TypeRegistry::getType)
79- .def (" registerType" , &TypeRegistry::registerType);
79+ .def (" registerType" , &TypeRegistry::registerType)
80+ .def (" flattenTypeHierarchy" , &TypeRegistry::flattenTypeHierarchy);
8081}
0 commit comments