chore: import upstream snapshot with attribution

python/ray/tune/tests/test_dependency.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python
+
+import sys
+
+import pytest
+
+import ray
+import ray.tune
+from ray.tune import register_trainable, run_experiments
+
+
+def f(config):
+    ray.tune.report(dict(timesteps_total=1))
+
+
+def test_dependency():
+    ray.init(num_cpus=2)
+
+    register_trainable("my_class", f)
+    run_experiments({"test": {"run": "my_class", "stop": {"training_iteration": 1}}})
+    assert "ray.rllib" not in sys.modules, "RLlib should not be imported"
+    assert "mlflow" not in sys.modules, "MLflow should not be imported"
+    ray.shutdown()
+
+
+if __name__ == "__main__":
+    sys.exit(pytest.main(["-v", __file__]))