task2 cleanup - final adjustments

nbody/utils/forces_basic.py

@@ -40,7 +40,7 @@ def n_body_forces(particles: np.ndarray, G: float = 1, softening: float = 0):
         r_adjusted[i] = 1
         num[i] = 0
 
-        f = - np.sum((num.T / r_adjusted**1.5).T, axis=0) * m_current
+        f = np.sum((num.T / r_adjusted**1.5).T, axis=0) * m_current
         forces[i] = f
 
         if i!= 0 and i % 5000 == 0:

nbody/utils/forces_cache.py

@@ -40,10 +40,9 @@ def kwargs_to_str(kwargs: dict):
     """
     base_str = ""
     for k, v in kwargs.items():
-        print(type(v))
         if type(v) == float:
             base_str += f"{k}_{v:.3f}"
-        elif type(v) == callable:
+        elif callable(v):
             base_str += f"{k}_{v.__name__}"
         else:
             base_str += f"{k}_{v}"