Black and isort

Author: p3jawors

abr_analyze/data_handler.py

@@ -229,7 +229,7 @@ def load(self, save_location, parameters=None, recursive=False):
                 tmp = bool(tmp)
             elif tmp.dtype == "object":
                 tmp = tmp.asstr()[()]
-                if tmp == 'None':
+                if tmp == "None":
                     tmp = None
                 # if not self.is_dataset(f"{save_location}/{key}"):
                 #     tmp = tmp.asstr()[()]

abr_analyze/data_logger.py

@@ -260,7 +260,9 @@ def find_experiments_that_match_constants(dat, saved_exp_hashes, const_params):
                     print(
                         f"{red}Got AttributeError on {param_key} who's value is:\n{data[param_key]}{endc}"
                     )
-                    print(f"{red}Or possibly from const params:\n{const_params[param_key]}{endc}")
+                    print(
+                        f"{red}Or possibly from const params:\n{const_params[param_key]}{endc}"
+                    )
                     raise e
             elif isinstance(data[param_key], dict):
                 raise NotImplementedError(
@@ -311,7 +313,9 @@ def get_common_experiments(
         saved_exp_hashes = find_experiments_that_match_constants(
             dat, saved_exp_hashes, const_params
         )
-        print(f"{green}{len(saved_exp_hashes)} experiments found with matching parameters{endc}")
+        print(
+            f"{green}{len(saved_exp_hashes)} experiments found with matching parameters{endc}"
+        )
         print(f"{green}{saved_exp_hashes}{endc}")
 
     # Get a dictionary of common values and a list of keys for differing values

abr_analyze/data_processor.py

@@ -209,7 +209,7 @@ def calc_cartesian_points(robot_config, q):
         The list of recorded joint angles used to transform link centers of
         mass and joint positions to cartesian coordinates
     """
-    assert robot_config is not None, 'robot_config must be provided'
+    assert robot_config is not None, "robot_config must be provided"
     if hasattr(robot_config, "xml_file"):
         mujoco_model = True
     else:
@@ -227,28 +227,28 @@ def calc_cartesian_points(robot_config, q):
         if not mujoco_model:
             # loop through the kinematic chain of joints
             for ii in range(0, robot_config.N_JOINTS):
-                joints_t_xyz.append(robot_config.Tx('joint%i'%ii, q=q_t))
-            joints_t_xyz.append(robot_config.Tx('EE', q=q_t))
+                joints_t_xyz.append(robot_config.Tx("joint%i" % ii, q=q_t))
+            joints_t_xyz.append(robot_config.Tx("EE", q=q_t))
 
             # loop through the kinematic chain of links
             for ii in range(0, robot_config.N_LINKS):
-                links_t_xyz.append(robot_config.Tx('link%i'%ii, q=q_t))
+                links_t_xyz.append(robot_config.Tx("link%i" % ii, q=q_t))
         else:
             # loop through the kinematic chain of joints
             for ii in range(0, robot_config.N_JOINTS):
-                joints_t_xyz.append(robot_config.Tx('joint%i'%ii, q=q_t, object_type="site"))
-            joints_t_xyz.append(robot_config.Tx('EE', q=q_t))
+                joints_t_xyz.append(
+                    robot_config.Tx("joint%i" % ii, q=q_t, object_type="site")
+                )
+            joints_t_xyz.append(robot_config.Tx("EE", q=q_t))
 
             # loop through the kinematic chain of links
-            for ii in range(0, robot_config.N_JOINTS+1):
+            for ii in range(0, robot_config.N_JOINTS + 1):
                 if ii == 0:
-                    name = 'base_link'
+                    name = "base_link"
                 else:
                     name = f"link{ii}"
                 links_t_xyz.append(robot_config.Tx(name, q=q_t))
 
-
-
         # append the cartesian coordinates of this time step to our list
         joints_xyz.append(joints_t_xyz)
         links_xyz.append(links_t_xyz)

abr_analyze/data_visualizer.py

@@ -6,9 +6,9 @@
 import matplotlib.colors as colors
 import matplotlib.pyplot as plt
 import numpy as np
+from matplotlib import rcParams
 from mpl_toolkits.axes_grid1 import make_axes_locatable
 from mpl_toolkits.mplot3d import axes3d  # pylint: disable=W0611
-from matplotlib import rcParams
 
 
 def plot_arm(
@@ -20,7 +20,7 @@ def plot_arm(
     joint_color="k",
     arm_color="k",
     title=None,
-    show_ground_collision=True
+    show_ground_collision=True,
 ):
     """
     Accepts joint, end-effector, and link COM cartesian locations, and an
@@ -56,26 +56,26 @@ def plot_arm(
     for xyz in joints_xyz:
         # plot joint location
         if xyz[2] <= 0.0:
-            marker = '*'
-            col = 'r'
-            s = rcParams['lines.markersize'] * 8
+            marker = "*"
+            col = "r"
+            s = rcParams["lines.markersize"] * 8
         else:
-            marker = 'o'
+            marker = "o"
             col = joint_color
-            s = rcParams['lines.markersize'] * 4
+            s = rcParams["lines.markersize"] * 4
 
         ax.scatter(xyz[0], xyz[1], xyz[2], c=col, marker=marker, s=s)
 
     for xyz in links_xyz:
         # plot link location
         if xyz[2] <= 0.0:
-            marker = 'x'
-            col = 'r'
-            s = rcParams['lines.markersize'] * 8
+            marker = "x"
+            col = "r"
+            s = rcParams["lines.markersize"] * 8
         else:
-            marker = 'o'
+            marker = "o"
             col = link_color
-            s = rcParams['lines.markersize'] * 4
+            s = rcParams["lines.markersize"] * 4
 
         ax.scatter(xyz[0], xyz[1], xyz[2], c=col, marker=marker, s=s)
 

abr_analyze/nengo/intercepts_scan.py

@@ -8,12 +8,12 @@
 import timeit
 
 import matplotlib.pyplot as plt
+import nengo
 import numpy as np
+from abr_control.controllers import signals
 
 import abr_analyze.nengo.network_utils as network_utils
-import nengo
 from abr_analyze.data_handler import DataHandler
-from abr_control.controllers import signals
 
 
 def run(

abr_analyze/nengo/network_utils.py

@@ -18,9 +18,8 @@
 import warnings
 
 import matplotlib.pyplot as plt
-import numpy as np
-
 import nengo
+import numpy as np
 from abr_control._vendor.nengolib.stats import ScatteredHypersphere
 from nengo.utils.matplotlib import rasterplot
 

abr_analyze/nengo_utils/tests/_test_network_utils.py

@@ -1,7 +1,7 @@
+import nengo
 import numpy as np
 import pytest
 
-import nengo
 from abr_analyze.nengo_utils import network_utils
 
 

abr_analyze/plotting/draw_arm.py

@@ -40,7 +40,7 @@ def plot(
         linestyle=None,
         label=None,
         title=None,
-        show_ground_collision=True
+        show_ground_collision=True,
     ):
         """
         Plots the parameters from save_location on the ax object
@@ -88,11 +88,11 @@ def plot(
         # plot our arm figure
         vis.plot_arm(
             ax=ax,
-            joints_xyz=data['joints_xyz'][step],
-            links_xyz=data['links_xyz'][step],
-            ee_xyz=data['ee_xyz'][step],
+            joints_xyz=data["joints_xyz"][step],
+            links_xyz=data["links_xyz"][step],
+            ee_xyz=data["ee_xyz"][step],
             title=title,
-            show_ground_collision=show_ground_collision
+            show_ground_collision=show_ground_collision,
         )
 
         return ax, [self.xlimit, self.ylimit, self.zlimit]

examples/gif_arm.py

@@ -11,12 +11,12 @@
 import os
 
 import matplotlib.pyplot as plt
+from abr_control.arms import jaco2
 from download_examples_db import check_exists as examples_db
 from mpl_toolkits.mplot3d import Axes3D
 
 from abr_analyze.paths import figures_dir
 from abr_analyze.plotting import Draw3dData, DrawArm, MakeGif
-from abr_control.arms import jaco2
 
 examples_db()
 gif = MakeGif()
@@ -49,7 +49,7 @@
     ax.set_xlim3d(-0.5, 0.5)
     ax.set_ylim3d(-0.5, 0.5)
     ax.set_zlim3d(0, 1)
-    ax.set_aspect('auto')
+    ax.set_aspect("auto")
 
     draw_3d.plot(
         ax=ax,

examples/learning_profile_from_saved.py

@@ -21,12 +21,12 @@
 import os
 
 import matplotlib.pyplot as plt
+from abr_control.controllers import signals
 from download_examples_db import check_exists as examples_db
 
 from abr_analyze import DataHandler
 from abr_analyze.nengo import network_utils
 from abr_analyze.paths import cache_dir, figures_dir
-from abr_control.controllers import signals
 
 examples_db()
 dat = DataHandler("abr_analyze_examples")

examples/plot_arm.py

@@ -11,11 +11,11 @@
 import os
 
 import matplotlib.pyplot as plt
+from abr_control.arms import jaco2
 from download_examples_db import check_exists as examples_db
 
 from abr_analyze.paths import figures_dir
 from abr_analyze.plotting import Draw3dData, DrawArm
-from abr_control.arms import jaco2
 
 examples_db()
 # the number of samples to interpolate our data to, set to None for no

examples/plot_arm_grid.py

@@ -7,12 +7,12 @@
 import matplotlib.gridspec as gridspec
 import matplotlib.pyplot as plt
 import numpy as np
+from abr_control.arms import jaco2
 from download_examples_db import check_exists as examples_db
 from mpl_toolkits.mplot3d import Axes3D
 
 from abr_analyze.paths import figures_dir
 from abr_analyze.plotting import Draw3dData, DrawArm, DrawCells
-from abr_control.arms import jaco2
 
 examples_db()
 interpolated_samples = 100

examples/plot_grid.py

@@ -4,10 +4,10 @@
 """
 import matplotlib.gridspec as gridspec
 import matplotlib.pyplot as plt
+from abr_control.arms import jaco2
 from download_examples_db import check_exists as examples_db
 
 from abr_analyze.plotting import Draw2dData, Draw3dData, DrawArm, DrawCells
-from abr_control.arms import jaco2
 
 examples_db()
 # the number of samples to interpolate our data to, set to None for no

examples/save_npz_to_hdf5.py

@@ -1,11 +1,12 @@
 import numpy as np
 from download_examples_db import check_exists as examples_db
+
 from abr_analyze.utils import npz_to_hdf5
 
 # TODO the name of the npz file you are sampling
 save_name = "test.npz"
 # this is the 'folder' in the database that the data will be saved to
-test_name =  'test_0000'
+test_name = "test_0000"
 
 examples_db()
 a = np.ones(11)
@@ -17,4 +18,4 @@
     db_name="abr_analyze_examples",
     save_location="my_converted_data/test1",
     overwrite=True,
-    )
+)