wehub-resource-sync
28 lines
609 B
Python
28 lines
609 B
Python
#!/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__]))
|