Pacify ci

PangeAI · Jun 3, 2024 · 06b54e9 · 06b54e9
1 parent 6d54170
commit 06b54e9
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/cudams/similarity/spectrum_similarity_functions.py b/cudams/similarity/spectrum_similarity_functions.py
@@ -460,10 +460,10 @@ def _kernel(
         #         break
 
     def kernel(
-        rspec: cuda.devicearray.DeviceNDArray,
-        qspec: cuda.devicearray.DeviceNDArray,
-        metadata: cuda.devicearray.DeviceNDArray,
-        out: cuda.devicearray.DeviceNDArray,
+        rspec,
+        qspec,
+        metadata,
+        out,
         stream: cuda.stream = None,
     ) -> None:
         """