minor issue fix with previous commit

PolyPhyHub · Oct 4, 2023 · e9919ea · e9919ea
1 parent 3a8d88f
commit e9919ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/experiments/jupyter/production/polyphy_3DDiscrete.py b/experiments/jupyter/production/polyphy_3DDiscrete.py
@@ -12,7 +12,7 @@ def __init__(self):
         self.rng = default_rng()
         self.ppInputData = PPInputData_3DDiscrete(self.input_file, self.rng)
         self.ppConfig.register_data(self.ppInputData)
-        ti.init(rch=ti.cpu if os.path.exists("/tmp/flag") else ti.gpu, device_memory_GB=4)
+        ti.init(arch=ti.cpu if os.path.exists("/tmp/flag") else ti.gpu, device_memory_GB=4)
         self.kernels = PPKernels()
         self.ppInternalData = PPInternalData_3DDiscrete(self.rng, self.kernels, self.ppConfig)