Remove workaround for huggingface/transformers#29032 (mlflow#11417)

Signed-off-by: harupy <17039389+harupy@users.noreply.github.com>
chenmoneygithub · Mar 18, 2024 · e65fc04 · e65fc04
1 parent 43e5cee
commit e65fc04
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/mlflow/ml-package-versions.yml b/mlflow/ml-package-versions.yml
@@ -527,6 +527,7 @@ transformers:
         # Avoid this patch: https://github.com/huggingface/transformers/pull/29032
         "4.38.0",
         "4.38.1",
+        "4.38.2",
       ]
     requirements:
       ">= 0.0.0": [
@@ -560,6 +561,7 @@ transformers:
         # Avoid this patch: https://github.com/huggingface/transformers/pull/29032
         "4.38.0",
         "4.38.1",
+        "4.38.2",
       ]
     requirements:
       ">= 0.0.0":

diff --git a/tests/transformers/test_transformers_autolog.py b/tests/transformers/test_transformers_autolog.py
@@ -22,12 +22,6 @@
 import mlflow
 
 
-# TODO: Remove this fixture once https://github.com/huggingface/transformers/pull/29096 is merged
-@pytest.fixture(autouse=True)
-def set_mlflow_tracking_uri_env_var(monkeypatch: pytest.MonkeyPatch):
-    monkeypatch.setenv("MLFLOW_TRACKING_URI", mlflow.get_tracking_uri())
-
-
 @pytest.fixture
 def iris_data():
     iris = sklearn.datasets.load_iris()