Update usage of moviepy

* Fix tests
polyaxon · Apr 25, 2023 · 61732d7 · 61732d7
1 parent aaeb87d
commit 61732d7
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/traceml/requirements/dev.txt b/traceml/requirements/dev.txt
@@ -3,7 +3,7 @@ ipython
 scikit-learn
 altair==4.1.0
 matplotlib
-moviepy==1.0.2
+moviepy==1.0.3
 plotly==4.6.0
 Pillow
 bokeh

diff --git a/traceml/tests/test_tracking/test_run_tracking.py b/traceml/tests/test_tracking/test_run_tracking.py
@@ -130,7 +130,7 @@ def test_dict_run_info(self):
             ("user", "project_bar", uid),
         )
 
-    @patch("polyaxon.managers.base.os.path.expanduser")
+    @patch("clipped.config.manager.os.path.expanduser")
     def test_run_init(self, expanduser):
         uid = uuid.uuid4().hex
         expanduser.return_value = tempfile.mkdtemp()
@@ -907,7 +907,7 @@ def test_log_data_video(self):
             is False
         )
         with patch("traceml.tracking.run.Run._log_has_events") as log_dashboard:
-            self.run.log_video(name="my_video", data=tensor_np(shape=(4, 3, 1, 8, 8)))
+            self.run.log_video(name="my_video", data=tensor_np(shape=(4, 3, 3, 8, 8)))
         assert log_dashboard.call_count == 1
         self.event_logger.flush()
         assert (

diff --git a/traceml/traceml/processors/events_processors/events_video_processors.py b/traceml/traceml/processors/events_processors/events_video_processors.py
@@ -91,9 +91,9 @@ def make_video(
 
     try:  # older version of moviepy
         if content_type == "gif":
-            clip.write_gif(asset_path, verbose=False, progress_bar=False)
+            clip.write_gif(asset_path, verbose=False, logger=None)
         else:
-            clip.write_videofile(asset_path, verbose=False, progress_bar=False)
+            clip.write_videofile(asset_path, verbose=False, logger=None)
     except TypeError:
         if content_type == "gif":
             clip.write_gif(asset_path, verbose=False)