Skip to content

Commit

Permalink
[HWORKS-1037][Append] Add support for downloading artifact files from…
Browse files Browse the repository at this point in the history
… deployment object
  • Loading branch information
javierdlrm committed Oct 17, 2024
1 parent 4f8604e commit 3ef94f8
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions python/hsml/engine/model_engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -251,14 +251,14 @@ def _save_model_from_local_or_hopsfs_mount(
from_hdfs_model_path=model_path.replace(
constants.MODEL_REGISTRY.HOPSFS_MOUNT_PREFIX, ""
),
to_model_files_path=model_instance.files_path,
to_model_files_path=model_instance.model_files_path,
keep_original_files=keep_original_files,
update_upload_progress=update_upload_progress,
)
else:
self._upload_local_model(
from_local_model_path=model_path,
to_model_files_path=model_instance.files_path,
to_model_files_path=model_instance.model_files_path,
update_upload_progress=update_upload_progress,
upload_configuration=upload_configuration,
)
Expand Down Expand Up @@ -367,7 +367,7 @@ def save(
if step["id"] == 0:
# Create folders
self._engine.mkdir(model_instance.version_path)
self._engine.mkdir(model_instance.files_path)
self._engine.mkdir(model_instance.model_files_path)
if step["id"] == 1:

def update_upload_progress(n_dirs=0, n_files=0, step=step):
Expand Down Expand Up @@ -404,7 +404,7 @@ def update_upload_progress(n_dirs=0, n_files=0, step=step):
): # check hdfs relative and absolute
self._copy_or_move_hopsfs_model(
from_hdfs_model_path=model_path,
to_model_files_path=model_instance.files_path,
to_model_files_path=model_instance.model_files_path,
keep_original_files=keep_original_files,
update_upload_progress=update_upload_progress,
)
Expand Down

0 comments on commit 3ef94f8

Please sign in to comment.