diff --git a/.doctrees/environment.pickle b/.doctrees/environment.pickle
index 33f48b18..b92d6c45 100644
Binary files a/.doctrees/environment.pickle and b/.doctrees/environment.pickle differ
diff --git a/.doctrees/index.doctree b/.doctrees/index.doctree
index 7a6fe5b9..2ae4737e 100644
Binary files a/.doctrees/index.doctree and b/.doctrees/index.doctree differ
diff --git a/_modules/cneuromax/fitting/deeplearning/utils/lightning.html b/_modules/cneuromax/fitting/deeplearning/utils/lightning.html
index 97d21866..5fbc4620 100644
--- a/_modules/cneuromax/fitting/deeplearning/utils/lightning.html
+++ b/_modules/cneuromax/fitting/deeplearning/utils/lightning.html
@@ -487,7 +487,7 @@ <h1>Source code for cneuromax.fitting.deeplearning.utils.lightning</h1><div clas
     <span class="k">else</span><span class="p">:</span>
         <span class="n">per_device_batch_size</span> <span class="o">=</span> <span class="n">datamodule</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">fixed_per_device_batch_size</span>
 
-    <span class="k">if</span> <span class="ow">not</span> <span class="n">datamodule</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">fixed_per_device_num_workers</span><span class="p">:</span>
+    <span class="k">if</span> <span class="n">datamodule</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">fixed_per_device_num_workers</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
         <span class="n">proposed_per_device_num_workers</span> <span class="o">=</span> <span class="n">find_good_per_device_num_workers</span><span class="p">(</span>
             <span class="n">datamodule</span><span class="o">=</span><span class="n">datamodule</span><span class="p">,</span>
             <span class="n">per_device_batch_size</span><span class="o">=</span><span class="n">per_device_batch_size</span><span class="p">,</span>