From d120c63feec64353340c173bd415a99646addb37 Mon Sep 17 00:00:00 2001 From: kmonte Date: Mon, 22 Jul 2024 16:18:52 -0700 Subject: [PATCH] Remove Env.get_base_dir() PiperOrigin-RevId: 654932257 --- tfx/orchestration/experimental/core/env.py | 7 ------- tfx/orchestration/experimental/core/env_test.py | 3 --- 2 files changed, 10 deletions(-) diff --git a/tfx/orchestration/experimental/core/env.py b/tfx/orchestration/experimental/core/env.py index 5ec0496cd6..565322ff64 100644 --- a/tfx/orchestration/experimental/core/env.py +++ b/tfx/orchestration/experimental/core/env.py @@ -43,10 +43,6 @@ def get_orchestration_options( ) -> orchestration_options.OrchestrationOptions: """Gets orchestration options for the pipeline.""" - @abc.abstractmethod - def get_base_dir(self) -> Optional[str]: - """Returns the base directory for the pipeline.""" - @abc.abstractmethod def label_and_tag_pipeline_run( self, mlmd_handle, pipeline_id, pipeline_run_id, labels, tags @@ -161,9 +157,6 @@ def get_orchestration_options( del pipeline return orchestration_options.OrchestrationOptions() - def get_base_dir(self) -> Optional[str]: - return None - def label_and_tag_pipeline_run( self, mlmd_handle, pipeline_id, pipeline_run_id, labels, tags ) -> None: diff --git a/tfx/orchestration/experimental/core/env_test.py b/tfx/orchestration/experimental/core/env_test.py index 4cd0b721c8..14971bb5a3 100644 --- a/tfx/orchestration/experimental/core/env_test.py +++ b/tfx/orchestration/experimental/core/env_test.py @@ -29,9 +29,6 @@ class _TestEnv(env.Env): def get_orchestration_options(self, pipeline): raise NotImplementedError() - def get_base_dir(self): - raise NotImplementedError() - def label_and_tag_pipeline_run( self, mlmd_handle, pipeline_id, pipeline_run_id, labels, tags ):