def test_collect_tensorflow_info(self): run_info = {} logger._collect_tensorflow_info(run_info) self.assertNotEqual(run_info["tensorflow_version"], {}) self.assertEqual(run_info["tensorflow_version"]["version"], tf.VERSION) self.assertEqual(run_info["tensorflow_version"]["git_hash"], tf.GIT_VERSION)
def test_collect_tensorflow_info(self): run_info = {} logger._collect_tensorflow_info(run_info) self.assertNotEqual(run_info["tensorflow_version"], {}) self.assertEqual(run_info["tensorflow_version"]["version"], tf.version.VERSION) self.assertEqual(run_info["tensorflow_version"]["git_hash"], tf.version.GIT_VERSION)