add CI env vars for CL config

Author: JLBegin

.github/workflows/ci.yaml

@@ -39,4 +39,7 @@ jobs:
         run: pip install tox
 
       - name: Run tests
+        env:
+          PTO_DEVICE_INDEX: 0
+          PTO_N_WORK_UNITS: 1024
         run: tox -e py${{ matrix.python-version }}

pytissueoptics/rayscattering/opencl/config/CLConfig.py

@@ -174,6 +174,19 @@ def _load(self):
         with open(OPENCL_CONFIG_PATH, "r") as f:
             self._config = json.load(f)
 
+        if os.getenv("PTO_DEVICE_INDEX") is not None:
+            try:
+                self.DEVICE_INDEX = int(os.getenv("PTO_DEVICE_INDEX"))
+            except ValueError:
+                raise ValueError(f"Invalid value for PTO_DEVICE_INDEX: {os.getenv('PTO_DEVICE_INDEX')}. Must be an "
+                                 f"integer.")
+        if os.getenv("PTO_N_WORK_UNITS") is not None:
+            try:
+                self.N_WORK_UNITS = int(os.getenv("PTO_N_WORK_UNITS"))
+            except ValueError:
+                raise ValueError(f"Invalid value for PTO_N_WORK_UNITS: {os.getenv('PTO_N_WORK_UNITS')}. Must be an "
+                                    f"integer.")
+
     def _assertExists(self):
         if not os.path.exists(OPENCL_CONFIG_PATH):
             warnings.warn("No OpenCL config file found. Creating a new one.")