Blacken

Author: ksunden

data_prototype/wrappers.py

@@ -170,6 +170,7 @@ def __setattr__(self, key, value):
         else:
             super().__setattr__(key, value)
 
+
 class LineWrapper(ProxyWrapper):
     _wrapped_class = _Line2D
     _privtized_methods = ("set_xdata", "set_ydata", "set_data", "get_xdata", "get_ydata", "get_data")
@@ -188,12 +189,21 @@ def draw(self, renderer):
     def _update_wrapped(self, data):
         self._wrapped_instance.set_data(data["x"], data["y"])
 
+
 class PathCollectionWrapper(ProxyWrapper):
     _wrapped_class = _PathCollection
     _privtized_methods = (
-            "set_facecolors", "set_edgecolors", "set_offsets", "set_sizes", "set_paths",
-            "get_facecolors", "get_edgecolors", "get_offsets", "get_sizes", "get_paths",
-            )
+        "set_facecolors",
+        "set_edgecolors",
+        "set_offsets",
+        "set_sizes",
+        "set_paths",
+        "get_facecolors",
+        "get_edgecolors",
+        "get_offsets",
+        "get_sizes",
+        "get_paths",
+    )
 
     def __init__(self, data: DataContainer, nus=None, /, **kwargs):
         super().__init__(data, nus)
@@ -216,7 +226,6 @@ def _update_wrapped(self, data):
         self._wrapped_instance.set_sizes(data["sizes"])
 
 
-
 class ImageWrapper(ProxyWrapper):
     _wrapped_class = _AxesImage
 

examples/lissajous.py

@@ -46,13 +46,13 @@ def query(
     ) -> Tuple[Dict[str, Any], Union[str, int]]:
         def next_time():
             cur_time = time.time()
-            cur_time = np.array([cur_time, cur_time-.1, cur_time-.2, cur_time-0.3])
+            cur_time = np.array([cur_time, cur_time - 0.1, cur_time - 0.2, cur_time - 0.3])
 
-            phase = 15*np.pi * (self.scale * cur_time % 60) / 150
+            phase = 15 * np.pi * (self.scale * cur_time % 60) / 150
             marker_obj = mmarkers.MarkerStyle("o")
             return {
-                "x": np.cos(5*phase),
-                "y": np.sin(3*phase),
+                "x": np.cos(5 * phase),
+                "y": np.sin(3 * phase),
                 "phase": phase[0],
                 "sizes": np.array([256]),
                 "paths": [marker_obj.get_path().transformed(marker_obj.get_transform())],
@@ -83,13 +83,13 @@ def update(frame, art):
 lw = PathCollectionWrapper(sot_c, offset_transform=ax.transData)
 ax.add_artist(lw)
 ax.add_artist(fc)
-#ax.set_xticks([])
-#ax.set_yticks([])
+# ax.set_xticks([])
+# ax.set_yticks([])
 ax.set_aspect(1)
 ani = FuncAnimation(
     fig,
     partial(update, art=(lw, fc)),
-    frames=60*15,
+    frames=60 * 15,
     interval=1000 / 100,
     # TODO: blitting does not work because wrappers do not inherent from Artist
     # blit=True,

examples/simple_scatter.py

@@ -25,23 +25,24 @@
 def update(frame, art):
     return art
 
+
 marker_obj = mmarkers.MarkerStyle("o")
 
 cont = ArrayContainer(
-    x = np.array([0,1,2]),
-    y = np.array([1,4,2]),
-    paths = np.array([marker_obj.get_path()]),
-    sizes = np.array([12]),
-    edgecolors = np.array(["k"]),
-    facecolors = np.array(["C3"]),
+    x=np.array([0, 1, 2]),
+    y=np.array([1, 4, 2]),
+    paths=np.array([marker_obj.get_path()]),
+    sizes=np.array([12]),
+    edgecolors=np.array(["k"]),
+    facecolors=np.array(["C3"]),
 )
 
 fig, ax = plt.subplots()
-ax.set_xlim(-.5, 2.5)
+ax.set_xlim(-0.5, 2.5)
 ax.set_ylim(0, 5)
 lw = PathCollectionWrapper(cont, offset_transform=ax.transData)
 ax.add_artist(lw)
-#ax.set_xticks([])
-#ax.set_yticks([])
-#ax.set_aspect(1)
+# ax.set_xticks([])
+# ax.set_yticks([])
+# ax.set_aspect(1)
 plt.show()