Use is_convertable_to_image|html in create_from

neptune-ai · Sep 19, 2022 · bdb5a00 · bdb5a00
1 parent 41bda63
commit bdb5a00
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neptune/new/types/atoms/file.py b/neptune/new/types/atoms/file.py
@@ -256,9 +256,9 @@ def as_pickle(obj) -> "File":
     def create_from(value) -> "File":
         if isinstance(value, str):
             return File(path=value)
-        elif is_pil_image(value) or is_matplotlib_figure(value):
+        elif File.is_convertable_to_image(value):
             return File.as_image(value)
-        elif is_plotly_figure(value) or is_altair_chart(value) or is_bokeh_figure(value):
+        elif File.is_convertable_to_html(value):
             return File.as_html(value)
         elif is_numpy_array(value):
             raise TypeError(