updated type hint of estimator ids to be FeatureSetId instead

NREL · May 29, 2024 · 3eeb68f · 3eeb68f
1 parent 655385b
commit 3eeb68f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nrel/routee/powertrain/core/model.py b/nrel/routee/powertrain/core/model.py
@@ -187,7 +187,7 @@ def to_file(self, file: Union[str, Path]):
 
     def visualize_features(
         self,
-        estimator_id: str,
+        estimator_id: FeatureSetId,
         n_samples: Optional[int] = 100,
         output_path: Optional[str] = None,
         return_predictions: Optional[bool] = False,
@@ -234,7 +234,7 @@ def visualize_features(
 
     def contour(
         self,
-        estimator_id: str,
+        estimator_id: FeatureSetId,
         x_feature: str,
         y_feature: str,
         n_samples: Optional[int] = 100,