Fix Python warning "is" with a literal

gyoto · Jan 17, 2024 · 38c25de · 38c25de
1 parent fa9c4f0
commit 38c25de
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/gyoto/util.py b/python/gyoto/util.py
@@ -167,7 +167,7 @@ def rayTrace(sc,
     else:
         raise TypeError('coord2dset must be a gyoto.core.Coord2dSet subclass or instance')
 
-    if isinstance(coord2dset, core.Grid) and scalars is 0 :
+    if isinstance(coord2dset, core.Grid) and scalars == 0 :
         dims=(ny, nx)
         array_double_fromnumpy1or2=core.array_double.fromnumpy2
         array_double_fromnumpy2or3=core.array_double.fromnumpy3
@@ -289,7 +289,7 @@ def rayTrace(sc,
     # Perform the actual ray-tracing
     sc.rayTrace(coord2dset, aop)
 
-    if scalars is 2:
+    if scalars == 2:
         for key in res:
             res[key]=res[key][0]