Commit 938d372e authored by Yanghan Wang's avatar Yanghan Wang Committed by Facebook GitHub Bot
Browse files

fix test_integration_tests_coco

Summary:
Pull Request resolved: https://github.com/facebookresearch/d2go/pull/671

> AttributeError: type object 'GeneralizedRCNNTask' has no attribute 'cleanup'

EZ fix

Reviewed By: rbasch

Differential Revision: D60400187

fbshipit-source-id: 25872f4928cf8851ff63e96311c12086e272d619
parent e6d115e9
...@@ -369,6 +369,10 @@ class DefaultTask(D2GoDataAPIMixIn, pl.LightningModule): ...@@ -369,6 +369,10 @@ class DefaultTask(D2GoDataAPIMixIn, pl.LightningModule):
cfg=cfg, dataset_name=dataset_name, output_folder=output_folder cfg=cfg, dataset_name=dataset_name, output_folder=output_folder
) )
@classmethod
def cleanup(cls) -> None:
pass
# --------------------------------------------------------------------------- # ---------------------------------------------------------------------------
# Hooks # Hooks
# --------------------------------------------------------------------------- # ---------------------------------------------------------------------------
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment