import tensorflow as tf

import tensorflow as tf

from garage.experiment import deterministic
from tests.fixtures.logger import NullOutput


class TfTestCase:
    def setup_method(self):
        self.sess = tf.compat.v1.Session()
        self.sess.__enter__()

    def teardown_method(self):
        self.sess.__exit__(None, None, None)
        self.sess.close()
        del self.sess
        gc.collect()
