style

Author: mbruno46

pyobs/core/data.py

@@ -28,6 +28,7 @@
 def is_int(x):
     return pyobs.is_type(x, pyobs.types.INT)
 
+
 def expand_data(data, idx, shape):
     v = np.prod(shape)
     tmp = None

pyobs/core/ndobs.py

@@ -424,9 +424,7 @@ def __setitem__(self, args, yobs):
             args = [args]
         else:
             args = [
-                [a]
-                if pyobs.is_type(a, pyobs.types.INT, slice, np.ndarray)
-                else a
+                [a] if pyobs.is_type(a, pyobs.types.INT, slice, np.ndarray) else a
                 for a in args
             ]
 
@@ -470,15 +468,15 @@ def rt(self, axis=None):
            >>> obs.rt().shape
            (10, 3)
         """
-        ## duplicate of pyobs.remove_tensor, to be eventually deprecated
+        # duplicate of pyobs.remove_tensor to be eventually deprecated
         Nd = len(self.shape)
         if axis is None:
             selection = [True] * Nd
         else:
             selection = [False] * Nd
             for a in pyobs.to_list(axis):
                 selection[a] = True
-                
+
         new_shape = []
         for mu in range(Nd):
             if (self.shape[mu] == 1) and (selection[mu] is True):
@@ -487,8 +485,7 @@ def rt(self, axis=None):
         if not new_shape:
             new_shape.append(1)
         return pyobs.reshape(self, tuple(new_shape))
-        
-            
+
     ##################################
     # overloaded basic math operations
 
@@ -689,6 +686,7 @@ class provides additional details for the automatic or manual
            >>> einfo = {'A': errinfo(Stau=3.0), 'B': errinfo(W=30)}
            >>> [v,e] = obsC.error(errinfo=einfo,plot=True)
         """
+
         def error_real(obs):
             [sigma, sigma_tot, _] = obs.error_core(errinfo, plot, pfile)
 
@@ -697,9 +695,13 @@ def error_real(obs):
                 if sum(h) > 1:
                     plot_piechart(obs.description, sigma, sigma_tot.real)
             return sigma_tot
-        
+
         if np.iscomplexobj(self.mean):
-            return [self.mean, np.sqrt(error_real(self.real())) + 1j*np.sqrt(error_real(self.imag()))]        
+            return [
+                self.mean,
+                np.sqrt(error_real(self.real()))
+                + 1j * np.sqrt(error_real(self.imag())),
+            ]
         return [self.mean, np.sqrt(error_real(self))]
 
     def error_breakdown(self, errinfo={}):

pyobs/default.py

@@ -24,11 +24,11 @@
 import time
 
 __all__ = [
-    "is_verbose", 
-    "set_verbose", 
-    "log_timer", 
-    "complex", 
-    "double", 
+    "is_verbose",
+    "set_verbose",
+    "log_timer",
+    "complex",
+    "double",
     "int",
 ]
 
@@ -53,6 +53,7 @@ def set_verbose(func, yesno=True):
         if func in verbose:
             verbose.remove(func)
 
+
 def log_timer(tag):
     def decorator(func):
         @functools.wraps(func)

pyobs/utils.py

@@ -164,14 +164,17 @@ def check_type(obj, s, *t):
     if c == len(t):
         raise TypeError(f"Unexpected type for {s} [{t}]")
 
+
 class types:
     INT = (int, numpy.int32, numpy.int64)
     FLOAT = (float, numpy.float32, numpy.float64)
     COMPLEX = (complex, numpy.complex64, numpy.complex128)
 
+
 def is_type(x, *args):
     return isinstance(x, args)
 
+
 def slice_to_range(sl, n):
     return list(range(n)[sl])
 
@@ -244,7 +247,8 @@ def core(string):
         return numpy.array(out)
     return core(data)
 
+
 def to_list(x):
-    if numpy.isdim(x)==0:
+    if numpy.isdim(x) == 0:
         return [x]
-    return list(x)
+    return list(x)