update

Author: deependujha

requirements/ci.txt

@@ -7,3 +7,4 @@ wget
 pkginfo ==1.12.1.2
 packaging <25.1
 tomlkit
+fastapi >=0.111.0  # to prevent `schema` error in `pydantic`

requirements/fabric/base.txt

@@ -6,3 +6,4 @@ fsspec[http] >=2022.5.0, <2025.11.0
 packaging >=20.0, <=25.0
 typing-extensions >4.5.0, <4.16.0
 lightning-utilities >=0.10.0, <0.16.0
+pytorch-lightning-enterprise >=0.0.1dev7

requirements/fabric/test.txt

@@ -11,4 +11,3 @@ click ==8.1.8; python_version < "3.11"
 click ==8.3.0; python_version > "3.10"
 tensorboardX >=2.6, <2.7.0  # todo: relax it back to `>=2.2` after fixing tests
 huggingface-hub
-pytorch-lightning-enterprise >=0.0.1dev7

requirements/pytorch/base.txt

@@ -9,3 +9,4 @@ torchmetrics >0.7.0, <1.9.0
 packaging >=20.0, <=25.0
 typing-extensions >4.5.0, <4.16.0
 lightning-utilities >=0.10.0, <0.16.0
+pytorch-lightning-enterprise >=0.0.1dev7

requirements/pytorch/test.txt

@@ -16,11 +16,10 @@ onnxruntime >=1.12.0, <1.24.0
 onnxscript >= 0.1.0, < 0.6.0
 psutil <7.1.2 # for `DeviceStatsMonitor`
 pandas >2.0, <2.4.0  # needed in benchmarks
-fastapi  # for `ServableModuleValidator`  # not setting version as re-defined in App
+fastapi >=0.111.0  # for `ServableModuleValidator`  # not setting version as re-defined in App
 uvicorn  # for `ServableModuleValidator`  # not setting version as re-defined in App
 
 tensorboard >=2.11, <2.21.0  # for `TensorBoardLogger`
 
 torch-tensorrt; platform_system == "Linux" and python_version >= "3.12"
 huggingface-hub
-pytorch-lightning-enterprise >=0.0.1dev7

tests/tests_fabric/graveyard/test_tpu.py

@@ -18,10 +18,6 @@ def test_graveyard_single_tpu(import_path, name, tpu_available):
     with pytest.deprecated_call(match="is deprecated"):
         cls(device)
 
-    # required to prevent env-var leakage
-    if hasattr(cls, "teardown"):
-        cls.teardown()
-
 
 @pytest.mark.parametrize(
     ("import_path", "name"),
@@ -42,8 +38,8 @@ def test_graveyard_no_device(import_path, name, tpu_available):
     module = import_module(import_path)
     cls = getattr(module, name)
     with pytest.deprecated_call(match="is deprecated"):
-        cls()
+        _instance = cls()
 
     # required to prevent env-var leakage
     if hasattr(cls, "teardown"):
-        cls.teardown()
+        cls.teardown(_instance)

tests/tests_pytorch/graveyard/test_tpu.py

@@ -18,10 +18,6 @@ def test_graveyard_single_tpu(import_path, name):
     with pytest.deprecated_call(match="is deprecated"), pytest.raises(ModuleNotFoundError, match="torch_xla"):
         cls(device)
 
-    # required to prevent env-var leakage
-    if hasattr(cls, "teardown"):
-        cls.teardown()
-
 
 @pytest.mark.parametrize(
     ("import_path", "name"),
@@ -43,7 +39,3 @@ def test_graveyard_no_device(import_path, name):
     cls = getattr(module, name)
     with pytest.deprecated_call(match="is deprecated"), pytest.raises(ModuleNotFoundError, match="torch_xla"):
         cls()
-
-    # required to prevent env-var leakage
-    if hasattr(cls, "teardown"):
-        cls.teardown()