import tensorflow.contrib.eager as tfe


import tensorflow as tf

import tensorflow.contrib.eager as tfe
from tensorflow.contrib.eager.python.examples.linear_regression import linear_regression


def device():
  return "/device:GPU:0" if tfe.num_gpus() > 0 else "/device:CPU:0"


class LinearRegressionTest(tf.test.TestCase):

  def setUp(self):
    super(LinearRegressionTest, self).setUp()
    self._tmp_logdir = tempfile.mkdtemp()
