def __init__( self, owner: str = None, project: str = None, run_uuid: str = None, client: RunClient = None, track_code: bool = True, track_env: bool = True, refresh_data: bool = False, artifacts_path: str = None, ): super().__init__( owner=owner, project=project, run_uuid=run_uuid, client=client, ) self.track_code = track_code self.track_env = track_env self._has_model = False self._has_events = False self._has_metrics = False self._has_tensorboard = False self._artifacts_path = None self._outputs_path = None self._event_logger = None self._resource_logger = None self._results = {} if (settings.CLIENT_CONFIG.is_managed and self.run_uuid) or artifacts_path: self.set_artifacts_path(artifacts_path) if ( self._artifacts_path and settings.CLIENT_CONFIG.is_managed and get_collect_artifact() ): self.set_run_event_logger() if get_collect_resources(): self.set_run_resource_logger() # no artifacts path is set, we use the temp path if not self._artifacts_path: self._artifacts_path = TEMP_RUN_ARTIFACTS self._outputs_path = TEMP_RUN_ARTIFACTS self._run = polyaxon_sdk.V1Run() if settings.CLIENT_CONFIG.is_offline: return if self._run_uuid and (refresh_data or settings.CLIENT_CONFIG.is_managed): self.refresh_data() # Track run env if settings.CLIENT_CONFIG.is_managed and self.track_env: self.log_env() if settings.CLIENT_CONFIG.is_managed: self._register_wait()
def __init__( self, owner=None, project=None, run_uuid=None, client=None, track_code=True, track_env=False, refresh_data=True, ): super().__init__( owner=owner, project=project, run_uuid=run_uuid, client=client, ) self.track_code = track_code self.track_env = track_env self._artifacts_path = None self._outputs_path = None self._event_logger = None self._resource_logger = None self._results = {} if settings.CLIENT_CONFIG.is_managed and self.run_uuid: self.set_run_event_path() if (self.artifacts_path and settings.CLIENT_CONFIG.is_managed and get_collect_artifact()): self.set_run_event_logger() if get_collect_resources(): self.set_run_resource_logger() self._run = polyaxon_sdk.V1Run() if settings.CLIENT_CONFIG.is_offline: return if self._run_uuid and refresh_data: self.refresh_data() # Track run env if settings.CLIENT_CONFIG.is_managed and self.track_env: self.log_run_env() self._register_wait()
def test_empty_collect_resources_path(self): settings.CLIENT_CONFIG.is_managed = True assert getters.get_collect_resources() is False
def test_get_collect_resources_return_false_out_cluster(self): settings.CLIENT_CONFIG.is_managed = False os.environ[POLYAXON_KEYS_COLLECT_RESOURCES] = "false" assert getters.get_collect_resources() is False