mirror of
https://github.com/vale981/ray
synced 2025-03-11 13:46:40 -04:00
60d4d5e1aa
* Remove all __future__ imports from RLlib. * Remove (object) again from tf_run_builder.py::TFRunBuilder. * Fix 2xLINT warnings. * Fix broken appo_policy import (must be appo_tf_policy) * Remove future imports from all other ray files (not just RLlib). * Remove future imports from all other ray files (not just RLlib). * Remove future import blocks that contain `unicode_literals` as well. Revert appo_tf_policy.py to appo_policy.py (belongs to another PR). * Add two empty lines before Schedule class. * Put back __future__ imports into determine_tests_to_run.py. Fails otherwise on a py2/print related error.
20 lines
517 B
Python
20 lines
517 B
Python
#!/usr/bin/env python
|
|
|
|
import os
|
|
import sys
|
|
|
|
os.environ["RLLIB_TEST_NO_TF_IMPORT"] = "1"
|
|
|
|
if __name__ == "__main__":
|
|
from ray.rllib.agents.a3c import A2CTrainer
|
|
assert "tensorflow" not in sys.modules, "TF initially present"
|
|
|
|
# note: no ray.init(), to test it works without Ray
|
|
trainer = A2CTrainer(
|
|
env="CartPole-v0", config={
|
|
"use_pytorch": True,
|
|
"num_workers": 0
|
|
})
|
|
trainer.train()
|
|
|
|
assert "tensorflow" not in sys.modules, "TF should not be imported"
|