Remove Env.get_base_dir()

PiperOrigin-RevId: 654932257
tensorflow · Jul 22, 2024 · d120c63 · d120c63
1 parent 2e66d21
commit d120c63
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 10 deletions.
diff --git 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
@@ -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
   ):