Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def _target_load(self, file_name: str) -> Any:
"""Select gokart load_function and load model.
Args:
file_name: Path to gokart's output file.
Returns:
Loaded data.
"""
file_path = self.client.to_absolute_path(file_name)
if file_path.endswith('.zip'):
tmp_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), os.path.abspath(self.tmp_path))
zip_client = gokart.zip_client_util.make_zip_client(file_path, tmp_path)
zip_client.unpack_archive()
load_function_path = os.path.join(tmp_path, 'load_function.pkl')
load_function = gokart.target.make_target(load_function_path).load()
model = load_function(os.path.join(tmp_path, 'model.pkl'))
shutil.rmtree(tmp_path)
return model
return gokart.target.make_target(file_path=file_path).load()
def __init__(self, file_path: str, temporary_directory: str, load_function, save_function) -> None:
self._zip_client = make_zip_client(file_path, temporary_directory)
self._temporary_directory = temporary_directory
self._save_function = save_function
self._load_function = load_function