Restructured folder structure

GoogleCloudPlatform · Mar 31, 2024 · e646091 · e646091
1 parent 54cc7be
commit e646091
Show file tree

Hide file tree

Showing 37 changed files with 596 additions and 751 deletions.
diff --git a/google_cloud_automlops/AutoMLOps.py b/google_cloud_automlops/AutoMLOps.py
@@ -61,17 +61,12 @@
     write_file
 )
 # Orchestration imports
-from google_cloud_automlops.orchestration.enums import (
+from google_cloud_automlops.orchestration.base import (
     Orchestrator,
     PipelineJobSubmitter
 )
-
-from google_cloud_automlops.orchestration.Component import Component
-from google_cloud_automlops.orchestration.Pipeline import Pipeline
-from google_cloud_automlops.orchestration.Services import Services
-from google_cloud_automlops.orchestration.kfp.KFPComponent import KFPComponent
-from google_cloud_automlops.orchestration.kfp.KFPPipeline import KFPPipeline
-from google_cloud_automlops.orchestration.kfp.KFPServices import KFPServices
+from google_cloud_automlops.orchestration.base import BaseComponent, BasePipeline, BaseServices
+from google_cloud_automlops.orchestration.kfp import KFPComponent, KFPPipeline, KFPServices
 
 # Provisioning imports
 from google_cloud_automlops.provisioning.pulumi import builder as PulumiBuilder
@@ -562,7 +557,7 @@ def my_function_one(input: str, output: Output[Model]):
             component,
             packages_to_install=packages_to_install)
     else:
-        components_dict[func.__name__] = Component(
+        components_dict[func.__name__] = BaseComponent(
             func=func,
             packages_to_install=packages_to_install
         )
@@ -603,7 +598,7 @@ def pipeline(bq_table: str,
             description=description)
     else:
         global pipeline_glob
-        pipeline_glob = Pipeline(func=func,
+        pipeline_glob = BasePipeline(func=func,
                                  name=name,
                                  description=description,
                                  comps_dict=components_dict)

diff --git a/google_cloud_automlops/orchestration/Component.py b/google_cloud_automlops/orchestration/Component.py
diff --git a/google_cloud_automlops/orchestration/Pipeline.py b/google_cloud_automlops/orchestration/Pipeline.py