From 20df1b16556be1671c64894b1197a4fa9ff541b4 Mon Sep 17 00:00:00 2001
From: Valeh Valiollah Pour Amiri <4193454+watiss@users.noreply.github.com>
Date: Tue, 16 Aug 2022 09:55:13 -0700
Subject: [PATCH] fix max_epochs in totalvi train

---
 scvi/model/_totalvi.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scvi/model/_totalvi.py b/scvi/model/_totalvi.py
index acf53ee28e..a0df889695 100644
--- a/scvi/model/_totalvi.py
+++ b/scvi/model/_totalvi.py
@@ -192,7 +192,7 @@ def __init__(
 
     def train(
         self,
-        max_epochs: Optional[int] = 400,
+        max_epochs: Optional[int] = None,
         lr: float = 4e-3,
         use_gpu: Optional[Union[str, int, bool]] = None,
         train_size: float = 0.9,