adding new classifier with scaling and preprocessing CI test

Author: Geoffrey-42

course4/week3-ungraded-labs/C4_W3_Lab_4_Github_Actions/app/main.py

@@ -9,7 +9,7 @@
 app = FastAPI(title="Predicting Wine Class with batching")
 
 # Open classifier in global scope
-with open("models/wine-95.pkl", "rb") as file:
+with open("models/wine-95-fixed.pkl", "rb") as file:
     clf = pickle.load(file)
 
 

course4/week3-ungraded-labs/C4_W3_Lab_4_Github_Actions/app/test_clf.py

@@ -1,6 +1,18 @@
 import pickle
 from main import clf
+from sklearn.pipeline import Pipeline
+from sklearn.preprocessing import StandardScaler
 
+def test_pipeline_and_scaler():
+    # Check if clf is an instance of sklearn.pipeline.Pipeline
+    isPipeline = isinstance(clf, Pipeline)
+    assert isPipeline
+    
+    if isPipeline:
+        # Check if first step of pipeline is an instance of
+        # sklearn.preprocessing.StandardScaler
+        firstStep = [v for v in clf.named_steps.values()][0]
+        assert isinstance(firstStep, StandardScaler)
 
 def test_accuracy():