add name to add_figure

visualDust · Dec 31, 2023 · 450da85 · 450da85
1 parent d2e00e8
commit 450da85
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/neetbox/client/apis/_image.py b/neetbox/client/apis/_image.py
@@ -169,6 +169,7 @@ def add_image(name: str, image, dataformats: str = None):
 
 
 def add_figure(
+    name,
     figure,
     close: Optional[bool] = True,
 ):
@@ -183,6 +184,6 @@ def add_figure(
         walltime: Override default walltime (time.time()) of event
     """
     if isinstance(figure, list):
-        add_image(image=figure_to_image(figure, close), dataformats="NCHW")
+        add_image(name=name, image=figure_to_image(figure, close), dataformats="NCHW")
     else:
-        add_image(image=figure_to_image(figure, close), dataformats="CHW")
+        add_image(name=name, image=figure_to_image(figure, close), dataformats="CHW")