Skip to content

Commit 8b1db01

Browse files
committed
Properly adjust h2o handling for Viya 3.5. #166
1 parent 951cd2c commit 8b1db01

File tree

1 file changed

+8
-16
lines changed

1 file changed

+8
-16
lines changed

src/sasctl/pzmm/write_score_code.py

Lines changed: 8 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -448,29 +448,21 @@ def _viya35_model_load(
448448

449449
if mojo_model:
450450
cls.score_code += (
451-
f'model_path = Path("/models/resources/viya/{model_id}'
452-
f'")\nwith gzip.open(model_path / "{model_file_name}'
453-
f'", "r") as fileIn, open(model_path / '
454-
f"\"{str(Path(model_file_name).with_suffix('.zip'))}\","
455-
f" \"wb\") as fileOut:\n{'':4}shutil.copyfileobj(fileIn,"
456-
" fileOut)\nos.chmod(model_path / "
457-
f"\"{str(Path(model_file_name).with_suffix('.zip'))}\""
458-
", 0o777)\nmodel = h2o.import_mojo(model_path / "
459-
f"\"{str(Path(model_file_name).with_suffix('.zip'))}\")"
460-
"\n\n"
451+
f"model = h2o.import_mojo(str(Path("
452+
f'"/models/resources/viya/{model_id}/{model_file_name}")))\n\n'
461453
)
462454
return (
463-
f"{'':8}model = h2o.import_mojo(model_path / \""
464-
f"{str(Path(model_file_name).with_suffix('.zip'))}\")"
455+
f"{'':8}model = h2o.import_mojo(str(Path("
456+
f'"/models/resources/viya/{model_id}/{model_file_name}")))'
465457
)
466458
elif binary_h2o_model:
467459
cls.score_code += (
468-
'model = h2o.load(Path("/models/resources/viya/'
469-
f'{model_id}/{model_file_name}"))\n\n'
460+
f'model = h2o.load(str(Path("/models/resources/viya/'
461+
f'{model_id}/{model_file_name}")))\n\n'
470462
)
471463
return (
472-
f' model = h2o.load(Path("/models/resources/viya/'
473-
f'{model_id}/{model_file_name}"))'
464+
f"{'':8}model = h2o.load(str(Path(\"/models/resources/viya/"
465+
f'{model_id}/{model_file_name}")))'
474466
)
475467
else:
476468
cls.score_code += (

0 commit comments

Comments
 (0)