Formatting

cms-patatrack · Jan 18, 2024 · 67ddb8e · 67ddb8e
1 parent 778da30
commit 67ddb8e
Showing 1 changed file with 13 additions and 10 deletions.
diff --git a/CLUEstering/alpaka/BindingModules/CLUEstering.py b/CLUEstering/alpaka/BindingModules/CLUEstering.py
@@ -573,28 +573,31 @@ def run_clue(self,
                                                     device_id)
         elif backend == "cpu tbb":
             if tbb_found:
-                cluster_id_is_seed = cpu_tbb.mainRun(self.dc_, self.rhoc, self.outlier, self.ppbin,
-                                                     self.clust_data.coords, self.clust_data.weight,
-                                                     self.kernel, self.clust_data.n_dim, block_size,
+                cluster_id_is_seed = cpu_tbb.mainRun(self.dc_, self.rhoc, self.outlier,
+                                                     self.ppbin, self.clust_data.coords,
+                                                     self.clust_data.weight, self.kernel,
+                                                     self.clust_data.n_dim, block_size,
                                                      device_id)
             else:
                 print("TBB module not found. Please re-compile the library and try again.")
 
         elif backend == "gpu cuda":
             if cuda_found:
-                cluster_id_is_seed = gpu_cuda.mainRun(self.dc_, float(self.rhoc), self.outlier, self.ppbin,
-                                                      self.clust_data.coords, self.clust_data.weight,
-                                                      self.kernel, self.clust_data.n_dim, block_size,
+                cluster_id_is_seed = gpu_cuda.mainRun(self.dc_, self.rhoc, self.outlier,
+                                                      self.ppbin, self.clust_data.coords,
+                                                      self.clust_data.weight, self.kernel,
+                                                      self.clust_data.n_dim, block_size,
                                                       device_id)
             else:
                 print("CUDA module not found. Please re-compile the library and try again.")
 
         elif backend == "gpu hip":
             if hip_found:
-                cluster_id_is_seed = gpu_hip.mainRun(self.dc_, float(self.rhoc), self.outlier, self.ppbin,
-                                                      self.clust_data.coords, self.clust_data.weight,
-                                                      self.kernel, self.clust_data.n_dim, block_size,
-                                                      device_id)
+                cluster_id_is_seed = gpu_hip.mainRun(self.dc_, self.rhoc, self.outlier,
+                                                     self.ppbin, self.clust_data.coords,
+                                                     self.clust_data.weight, self.kernel,
+                                                     self.clust_data.n_dim, block_size,
+                                                     device_id)
             else:
                 print("HIP module not found. Please re-compile the library and try again.")