diff --git a/nortech/derivers/__init__.py b/nortech/derivers/__init__.py index cd4b602..8711081 100644 --- a/nortech/derivers/__init__.py +++ b/nortech/derivers/__init__.py @@ -10,6 +10,7 @@ visualize_deriver, visualize_deriver_schema, ) +from nortech.derivers.services import operators as operators from nortech.derivers.services.physical_units import get_physical_quantity from nortech.derivers.values import physical_units from nortech.derivers.values.instance import ( @@ -65,4 +66,5 @@ def run_deriver_locally( "InputField", "OutputField", "get_physical_quantity", + "operators", ] diff --git a/tests/integration/derivers/test_operators.py b/tests/integration/derivers/test_operators.py index 4b8be08..9a4398e 100644 --- a/tests/integration/derivers/test_operators.py +++ b/tests/integration/derivers/test_operators.py @@ -7,8 +7,8 @@ from bytewax.dataflow import Dataflow from bytewax.testing import TestingSink, TestingSource, run_main -from nortech.derivers.services import operators as internal_op -from nortech.derivers.values.schema import DeriverInputSchema +from nortech.derivers import DeriverInputSchema +from nortech.derivers import operators as internal_op def test_key_all():