diff --git a/transforms/universal/doc_id/Makefile b/transforms/universal/doc_id/Makefile index 1db88041b..6d7dbfbf0 100644 --- a/transforms/universal/doc_id/Makefile +++ b/transforms/universal/doc_id/Makefile @@ -19,12 +19,12 @@ run-cli-spark-sample: make venv source venv/bin/activate && \ $(PYTHON) -m dpk_$(TRANSFORM_NAME).spark.transform \ - --run_locally True --data_local_config "{ 'input_folder' : '../test-data/input', 'output_folder' : '../output'}" \ + --data_local_config "{ 'input_folder' : 'test-data/input', 'output_folder' : 'output'}" \ --doc_id_int True run-cli-ray-sample: make venv source venv/bin/activate && \ $(PYTHON) -m dpk_$(TRANSFORM_NAME).ray.transform \ - --run_locally True --data_local_config "{ 'input_folder' : '../test-data/input', 'output_folder' : '../output'}" \ + --run_locally True --data_local_config "{ 'input_folder' : 'test-data/input', 'output_folder' : 'output'}" \ --doc_id_int True diff --git a/transforms/universal/doc_id/dpk_doc_id/transform_python.py b/transforms/universal/doc_id/dpk_doc_id/transform_python.py index 4dd5b4c6f..b8b886227 100644 --- a/transforms/universal/doc_id/dpk_doc_id/transform_python.py +++ b/transforms/universal/doc_id/dpk_doc_id/transform_python.py @@ -70,7 +70,7 @@ def apply_input_params(self, args: Namespace) -> bool: return super().apply_input_params(args=args) -class DocIDRuntime(DefaultPythonTransformRuntime): +class DocIDPythonRuntime(DefaultPythonTransformRuntime): """ Exact dedup runtime support """ @@ -110,7 +110,7 @@ class DocIDPythonTransformRuntimeConfiguration(PythonTransformRuntimeConfigurati def __init__(self): super().__init__( transform_config=DocIDTransformConfiguration(), - runtime_class=DocIDRuntime, + runtime_class=DocIDPythonRuntime, )