diff --git a/transforms/universal/hap/python/src/hap_local_python.py b/transforms/universal/hap/python/src/hap_local_python.py index 2dfe8ab2c9..ca5beb144a 100644 --- a/transforms/universal/hap/python/src/hap_local_python.py +++ b/transforms/universal/hap/python/src/hap_local_python.py @@ -17,14 +17,9 @@ from data_processing.utils import ParamsUtils from hap_transform_python import HAPPythonTransformConfiguration -# create parameters -# input_folder = os.path.abspath(os.path.join(os.path.dirname(__file__), "../test-data/input")) -# output_folder = os.path.abspath(os.path.join(os.path.dirname(__file__), "../output")) - input_folder = os.getenv('INPUT_FOLDER', os.path.abspath(os.path.join(os.path.dirname(__file__), "../test-data/input"))) output_folder = os.getenv('OUTPUT_FOLDER', os.path.abspath(os.path.join(os.path.dirname(__file__), "../output")) ) -print("PATHS: ", input_folder, output_folder) local_conf = { "input_folder": input_folder, @@ -39,7 +34,6 @@ "runtime_code_location": ParamsUtils.convert_to_ast(code_location), } - hap_params = { "model_name_or_path": os.getenv('MODEL_NAME_OR_PATH', 'ibm-granite/granite-guardian-hap-38m'), "annotation_column": os.getenv('ANNOTATION_COLUMN', "hap_score"), @@ -52,8 +46,7 @@ if __name__ == "__main__": # Set the simulated command line args sys.argv = ParamsUtils.dict_to_req(d=params | hap_params) - print("PARAMs: ", hap_params) # create launcher launcher = PythonTransformLauncher(runtime_config=HAPPythonTransformConfiguration()) # Launch the ray actor(s) to process the input - launcher.launch() \ No newline at end of file + launcher.launch()