removed the profiler utils

python/tests/graph/test_builder.py

@@ -35,16 +35,11 @@
 
 from ..tests import SparkDLTestCase
 from ..transformers.image_utils import _getSampleJPEGDir, getSampleImagePathsDF
-from ..utils import do_cprofile
 
 
 class GraphFunctionWithIsolatedSessionTest(SparkDLTestCase):
 
     def test_tf_consistency(self):
-        self._test_tf_consistency()
-
-    @do_cprofile
-    def _test_tf_consistency(self):
         """ Should get the same graph as running pure tf """
 
         x_val = 2702.142857
@@ -75,7 +70,6 @@ def _test_tf_consistency(self):
         # should be the same as that in the one exported directly from TensorFlow session
         self.assertEqual(str(gfn.graph_def), str(gdef_ref))
 
-    @do_cprofile
     def test_get_graph_elements(self):
         """ Fetching graph elements by names and other graph elements """
 
@@ -94,7 +88,6 @@ def test_get_graph_elements(self):
             self.assertEqual(tfx.tensor_name(z, g), "z:0")
             self.assertEqual(tfx.tensor_name(x, g), "x:0")
 
-    @do_cprofile
     def test_import_export_graph_function(self):
         """ Function import and export must be consistent """
 
@@ -111,12 +104,8 @@ def test_import_export_graph_function(self):
         self.assertEqual(gfn_tgt.output_names, gfn_ref.output_names)
         self.assertEqual(str(gfn_tgt.graph_def), str(gfn_ref.graph_def))
 
-    def test_keras_consistency(self):
-        self._test_keras_consistency()
-        assert False
 
-    @do_cprofile
-    def _test_keras_consistency(self):
+    def test_keras_consistency(self):
         """ Exported model in Keras should get same result as original """
 
         img_fpaths = glob(os.path.join(_getSampleJPEGDir(), '*.jpg'))

python/tests/utils/__init__.py

@@ -12,22 +12,3 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-
-import cProfile
-
-"""
-This function was copied from here:
-https://zapier.com/engineering/profiling-python-boss/
-"""
-def do_cprofile(func):
-    def profiled_func(*args, **kwargs):
-        profile = cProfile.Profile()
-        try:
-            profile.enable()
-            result = func(*args, **kwargs)
-            profile.disable()
-            return result
-        finally:
-            profile.print_stats()
-            profile.dump_stats("/tmp/stats.txt")
-    return profiled_func