#!/usr/bin/env python

import os
import sys
import pytest


@pytest.mark.skip(reason="Upstream change make it failed.")
def test_dependency_torch():
    # Do not import torch for testing purposes.
    os.environ["RLLIB_TEST_NO_TORCH_IMPORT"] = "1"

    from ray.rllib.agents.a3c import A2CTrainer
    assert "torch" not in sys.modules, \
        "Torch initially present, when it shouldn't."

    # note: no ray.init(), to test it works without Ray
    trainer = A2CTrainer(
        env="CartPole-v0", config={
            "framework": "tf",
            "num_workers": 0
        })
    trainer.train()

    assert "torch" not in sys.modules, "Torch should not be imported"

    # Clean up.
    del os.environ["RLLIB_TEST_NO_TORCH_IMPORT"]

    print("ok")


if __name__ == "__main__":
    sys.exit(pytest.main(["-sv", __file__]))