Python _collect_tensorflow_info Exemples

Exemple #1

Fichier : logger_test.py Projet : gustavz/tf_models

 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)

Exemple #2

Fichier : logger_test.py Projet : Exscotticus/models

 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)