Passing max_epochs=None to SCVI.train throws an error.
File "/usr/local/lib/python3.8/site-packages/scib/integration.py", line 280, in scanvi
vae = scvi(adata, batch, hvg, return_model=True, max_epochs=n_epochs_scVI)
File "/usr/local/lib/python3.8/site-packages/scib/integration.py", line 242, in scvi
vae.train(train_size=1.0, max_epochs=max_epochs)
File "/usr/local/lib/python3.8/site-packages/scvi/model/base/_training_mixin.py", line 142, in train
return runner()
File "/usr/local/lib/python3.8/site-packages/scvi/train/_trainrunner.py", line 82, in __call__
self.trainer.fit(self.training_plan, self.data_splitter)
File "/usr/local/lib/python3.8/site-packages/scvi/train/_trainer.py", line 188, in fit
super().fit(*args, **kwargs)
File "/usr/local/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 696, in fit
self._call_and_handle_interrupt(
File "/usr/local/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 650, in _call_and_handle_interrupt
return trainer_fn(*args, **kwargs)
File "/usr/local/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 735, in _fit_impl
results = self._run(model, ckpt_path=self.ckpt_path)
File "/usr/local/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 1166, in _run
results = self._run_stage()
File "/usr/local/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 1252, in _run_stage
return self._run_train()
File "/usr/local/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 1283, in _run_train
self.fit_loop.run()
File "/usr/local/lib/python3.8/site-packages/pytorch_lightning/loops/loop.py", line 190, in run
if self.skip:
File "/usr/local/lib/python3.8/site-packages/pytorch_lightning/loops/fit_loop.py", line 194, in skip
return self.done or self.trainer.limit_train_batches == 0
File "/usr/local/lib/python3.8/site-packages/pytorch_lightning/loops/fit_loop.py", line 165, in done
assert isinstance(self.max_epochs, int)
AssertionError
Passing
max_epochs=None
toSCVI.train
throws an error.