diff --git a/tests/unit_tests/explainer/test_smart_explainer.py b/tests/unit_tests/explainer/test_smart_explainer.py index 97269c09..76db4cc1 100644 --- a/tests/unit_tests/explainer/test_smart_explainer.py +++ b/tests/unit_tests/explainer/test_smart_explainer.py @@ -1,6 +1,7 @@ """ Unit test for smart explainer """ + import os import sys import types @@ -633,14 +634,14 @@ def test_load_1(self): temp, xpl = init_sme_to_pickle_test() current = Path(path.abspath(__file__)).parent.parent.parent - if str(sys.version)[0:3] == "3.8": - pkl_file = path.join(current, "data/xpl_to_load_38.pkl") - elif str(sys.version)[0:3] == "3.9": + if str(sys.version)[0:3] == "3.9": pkl_file = path.join(current, "data/xpl_to_load_39.pkl") elif str(sys.version)[0:4] == "3.10": pkl_file = path.join(current, "data/xpl_to_load_310.pkl") elif str(sys.version)[0:4] == "3.11": pkl_file = path.join(current, "data/xpl_to_load_311.pkl") + elif str(sys.version)[0:4] == "3.12": + pkl_file = path.join(current, "data/xpl_to_load_312.pkl") else: raise NotImplementedError diff --git a/tests/unit_tests/utils/test_load_smartpredictor.py b/tests/unit_tests/utils/test_load_smartpredictor.py index 4ce82b2e..72af76f0 100644 --- a/tests/unit_tests/utils/test_load_smartpredictor.py +++ b/tests/unit_tests/utils/test_load_smartpredictor.py @@ -1,6 +1,7 @@ """ Unit test smart predictor """ + import sys import unittest from os import path @@ -29,12 +30,12 @@ def test_load_smartpredictor_1(self): current = Path(path.abspath(__file__)).parent.parent.parent if str(sys.version)[0:4] == "3.10": pkl_file = path.join(current, "data/predictor_to_load_310.pkl") - elif str(sys.version)[0:3] == "3.8": - pkl_file = path.join(current, "data/predictor_to_load_38.pkl") elif str(sys.version)[0:3] == "3.9": pkl_file = path.join(current, "data/predictor_to_load_39.pkl") elif str(sys.version)[0:4] == "3.11": pkl_file = path.join(current, "data/predictor_to_load_311.pkl") + elif str(sys.version)[0:4] == "3.12": + pkl_file = path.join(current, "data/predictor_to_load_312.pkl") else: raise NotImplementedError