chore: get connection and target

rudderlabs · Nov 15, 2024 · 7031251 · 7031251
1 parent 5af2167
commit 7031251
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/predictions/profiles_mlcorelib/py_native/profiles_tutorial/database_manager.py b/src/predictions/profiles_mlcorelib/py_native/profiles_tutorial/database_manager.py
@@ -24,6 +24,9 @@ def __init__(
         self.io = io_handler
         self.fast_mode = fast_mode
 
+    def get_connection_and_target(self) -> Tuple[str, str]:
+        return self.client.get_connection_and_target()
+
     def get_qualified_name(self, table: str) -> str:
         """Returns the fully qualified name of the table"""
         return f"{self.db}.{self.schema}.{table}"

diff --git a/src/predictions/profiles_mlcorelib/py_native/profiles_tutorial/tutorial.py b/src/predictions/profiles_mlcorelib/py_native/profiles_tutorial/tutorial.py
@@ -41,7 +41,7 @@ def run(self, client: BaseClient):
         new_table_names = self.upload_sample_data()
         relevant_tables = self.find_relevant_tables(new_table_names)
 
-        conn_name, target = client.get_connection_and_target()
+        conn_name, target = self.db_manager.get_connection_and_target()
         self.display_about_project_files(conn_name)
         # pb_project.yaml
         entity_name, id_types, id_graph_model = self.generate_pb_project(conn_name)