Fix mypy

Author: tomwhite

cubed/core/plan.py

@@ -681,7 +681,8 @@ def visualize(
                 stacks.append(stack_summaries)
         # add current stack info
         # go back one in the stack to the caller of 'visualize'
-        frame = inspect.currentframe().f_back
+        frame = inspect.currentframe()
+        frame = frame.f_back if frame is not None else frame
         stack_summaries = extract_stack_summaries(frame, limit=10)
         stacks.append(stack_summaries)
         array_display_names = extract_array_names_from_stack_summaries(stacks)

cubed/diagnostics/widgets/plan.py

@@ -42,9 +42,9 @@ class PlanWidget(anywidget.AnyWidget):
     selected_node = traitlets.Unicode().tag(sync=True)
     _summary_html = traitlets.Unicode().tag(sync=True)
     _node_html_dict = traitlets.Dict({}).tag(sync=True)
-    _cytoscape_elements = traitlets.List([]).tag(sync=True)
+    _cytoscape_elements = traitlets.List([]).tag(sync=True)  # type: ignore[var-annotated]
     _cytoscape_layout = traitlets.Dict({}).tag(sync=True)
-    _cytoscape_style = traitlets.List([]).tag(sync=True)
+    _cytoscape_style = traitlets.List([]).tag(sync=True)  # type: ignore[var-annotated]
     _cytoscape_options = traitlets.Dict({}).tag(sync=True)
 
 

setup.cfg

@@ -9,6 +9,8 @@ allow_redefinition = True
 ignore_errors = True
 
 # Most of the numerical computing stack doesn't have type annotations yet.
+[mypy-anywidget.*]
+ignore_missing_imports = True
 [mypy-apache_beam.*]
 ignore_missing_imports = True
 [mypy-aiostream.*]
@@ -31,10 +33,12 @@ ignore_missing_imports = True
 ignore_missing_imports = True
 [mypy-icechunk.*]
 ignore_missing_imports = True
-[mypy-lithops.*]
+[mypy-ipywidgets.*]
 ignore_missing_imports = True
 [mypy-IPython.*]
 ignore_missing_imports = True
+[mypy-lithops.*]
+ignore_missing_imports = True
 [mypy-memray.*]
 ignore_missing_imports = True
 [mypy-modal.*]