LIN-740: Reset configurations (LineaLabs#870)

Some MLflow tests require setting some configuration flags using  which are not reset by default, which leads to subsequent tests failing due to unexpected configuration. Hence, for such tests, we reset the  as well as the env variables to tackle the bug.

tests/conftest.py

@@ -187,6 +187,8 @@ def __call__(
                     f"lineapy.{save.__name__}({artifact}, {repr(artifact)})\n"
                 )
 
+        # Saving old env variables to reset after test is executed
+        old_os_environ = dict(os.environ)
         # These temp filenames are unique per test function.
         # If `execute` is called twice in a test, it will overwrite the
         # previous paths
@@ -247,6 +249,13 @@ def __call__(
         new_executor.execute_graph(tracer.graph)
         os.chdir(current_working_dir)
 
+        # Reset environment variables to state before the user code was executed
+        # so as to not affect subsequently executed tests.
+        for key in os.environ.keys():
+            del os.environ[key]
+        for key in old_os_environ:
+            os.environ[key] = old_os_environ[key]
+
         return tracer
 
 

tests/test_mlflow.py

@@ -45,6 +45,8 @@ def test_save_get_delete_to_mlflow(execute):
     deleted_clf = lineapy.get('clf', version=art.version)
 except:
     deleted_clf = None
+lineapy.options.set('mlflow_tracking_uri',None)
+lineapy.options.set('mlflow_registry_uri',None)
 """
     res = execute(code, snapshot=False)
 

tests/unit/utils/test_config_mlflow.py

@@ -37,3 +37,7 @@ def test_mlflow_console_config():
     clean_lineapy_env_var()
     for k, v in existing_lineapy_env.items():
         os.environ[k] = v
+
+    # Reset MLFlow options to not affect subsequent tests
+    options.set("mlflow_tracking_uri", None)
+    options.set("default_ml_models_storage_backend", None)