Add types for _run

z3z1ma · Sep 30, 2023 · 01923e3 · 01923e3
1 parent 13e4d56
commit 01923e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/dbt_osmosis/core/osmosis.py b/src/dbt_osmosis/core/osmosis.py
@@ -782,7 +782,7 @@ def get_column_sets(
         ]
         return missing_columns, undocumented_columns, extra_columns
 
-    def _run(self, unique_id, node, schema_map, force_inheritance=False):
+    def _run(self, unique_id: str, node: ManifestNode, schema_map: Dict[str, SchemaFileLocation], force_inheritance: bool=False):
         try:
             with self.mutex:
                 logger().info(":point_right: Processing model: [bold]%s[/bold]", unique_id)