Browse Source

exclusion in test_tf

John Welsh 7 years ago
parent
commit
5ee4652fbd
1 changed files with 4 additions and 0 deletions
  1. 4 0
      scripts/test_tf.py

+ 4 - 0
scripts/test_tf.py

@@ -22,6 +22,10 @@ if __name__ == '__main__':
 
     with open(TEST_OUTPUT_PATH, 'w') as test_f:
         for net_name, net_meta in NETS.items():
+
+            if 'exclude' in net_meta.keys() and net_meta['exclude'] is True:
+                continue
+
             print("Testing %s" % net_name)
 
             with open(net_meta['frozen_graph_filename'], 'rb') as f: