BugFix: Python Profiler (#589)

awslabs · Apr 16, 2022 · a3ef5d6 · a3ef5d6
1 parent cdac463
commit a3ef5d6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/smdebug/_version.py b/smdebug/_version.py
@@ -1 +1 @@
-__version__ = "1.0.15"
+__version__ = "1.0.16"
diff --git a/tests/profiler/core/test_python_profiler.py b/tests/profiler/core/test_python_profiler.py
@@ -117,6 +117,8 @@ def upload_files(args):
 
     pool = ThreadPool(processes=10)
     pool.map(upload_files, filenames)
+    pool.close()
+    pool.join()
 
 
 def _validate_analysis(profiler_name, stats, expected_functions):