fixed unit tests

Author: elizjo

tests/unitary/with_extras/evaluator/test_evaluator.py

@@ -6,22 +6,23 @@
 """
 Contains tests for ads.evaluations.evaluator
 """
-import pytest
-import unittest
 import tempfile
-from ads.evaluations.evaluator import Evaluator
+import unittest
 
-from sklearn.datasets import make_regression
-from sklearn.datasets import make_classification
-from sklearn.model_selection import train_test_split
+import pandas as pd
+import pytest
 from lightgbm import LGBMClassifier, LGBMRegressor
-from sklearn.tree import DecisionTreeClassifier
+from sklearn.datasets import make_classification, make_regression
 from sklearn.linear_model import LinearRegression
-import pandas as pd
+from sklearn.model_selection import train_test_split
+from sklearn.tree import DecisionTreeClassifier
+
 from ads.common.model_metadata import UseCaseType
+from ads.evaluations.evaluator import Evaluator
 from ads.model.framework.lightgbm_model import LightGBMModel
 from ads.model.framework.sklearn_model import SklearnModel
 
+DEFAULT_PYTHON_VERSION = "3.12"
 
 def test_model_types():
     with pytest.raises(ValueError):
@@ -39,6 +40,7 @@ def test_pandas_input():
     model.prepare(
         inference_conda_env="generalml_p38_cpu_v1",
         training_conda_env="generalml_p38_cpu_v1",
+        inference_python_version=DEFAULT_PYTHON_VERSION,
         X_sample=X,
         y_sample=y,
         use_case_type=UseCaseType.MULTINOMIAL_CLASSIFICATION,
@@ -100,6 +102,7 @@ def train_eval_model(self, data, model, use_case):
         my_model.prepare(
             inference_conda_env="generalml_p38_cpu_v1",
             training_conda_env="generalml_p38_cpu_v1",
+            inference_python_version=DEFAULT_PYTHON_VERSION,
             X_sample=X_test,
             y_sample=y_test,
             use_case_type=use_case,

tests/unitary/with_extras/model/test_model_metadata_mixin.py

@@ -3,20 +3,22 @@
 # Copyright (c) 2022, 2023 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
+import os
+import shutil
 from unittest.mock import patch
+
 import numpy as np
 import pytest
+import sklearn
+import xgboost
 from sklearn import datasets, linear_model
 
-from ads.model.generic_model import GenericModel
+from ads.feature_engineering.schema import Schema
 from ads.model.framework.sklearn_model import SklearnModel
 from ads.model.framework.xgboost_model import XGBoostModel
-from ads.feature_engineering.schema import Schema
-import sklearn
-import os
-import shutil
-import xgboost
+from ads.model.generic_model import GenericModel
 
+DEFAULT_PYTHON_VERSION = "3.12"
 
 class TestMetadataMixin:
     def setup_method(cls):
@@ -49,6 +51,7 @@ def test_metadata_generic_model(self):
         model.prepare(
             inference_conda_env="dataexpl_p37_cpu_v3",
             namespace="ociodscdev",
+            inference_python_version=DEFAULT_PYTHON_VERSION,
             model_file_name="model.joblib",
             force_overwrite=True,
         )