Fix pytorch-lightning dev testing (#4398)
Fix pytorch-lightning dev testing (#4398)

* only define on_train_epoch_end when pl.__version__ > 1.4.0dev

Signed-off-by: harupy <3b81b8b110f2244bdbf0f43efa472935c395c613@users.noreply.github.com>

* comment

Signed-off-by: harupy <3b81b8b110f2244bdbf0f43efa472935c395c613@users.noreply.github.com>

* fix comment

Signed-off-by: harupy <3b81b8b110f2244bdbf0f43efa472935c395c613@users.noreply.github.com>

* remove blank line

Signed-off-by: harupy <3b81b8b110f2244bdbf0f43efa472935c395c613@users.noreply.github.com>
mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow