Fix bug with from_argparse and finetune

CHANGELOG.md

@@ -4,6 +4,12 @@ All notable changes to this project will be documented in this file.
 
 The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
+## [0.3.2] - 2021-06-08
+
+### Fixed
+
+- Fixed a bug where `flash.Trainer.from_argparse_args` + `finetune` would not work ([#382](https://github.com/PyTorchLightning/lightning-flash/pull/382))
+
 ## [0.3.1] - 2021-06-08
 
 ### Added

flash/core/trainer.py

@@ -202,4 +202,4 @@ def add_argparse_args(cls, *args, **kwargs) -> ArgumentParser:
     def from_argparse_args(cls, args: Union[Namespace, ArgumentParser], **kwargs) -> 'Trainer':
         # the lightning trainer implementation does not support subclasses.
         # context: https://github.com/PyTorchLightning/lightning-flash/issues/342#issuecomment-848892447
-        return from_argparse_args(PlTrainer, args, **kwargs)
+        return from_argparse_args(Trainer, args, **kwargs)

tests/core/test_trainer.py

@@ -126,3 +126,4 @@ def test_from_argparse_args():
     args = parser.parse_args(['--max_epochs=200'])
     trainer = Trainer.from_argparse_args(args)
     assert trainer.max_epochs == 200
+    assert isinstance(trainer, Trainer)