From 0c5f9592a8f801e53a7706c8e4d0b663e92399b5 Mon Sep 17 00:00:00 2001 From: Om Doiphode Date: Wed, 6 Dec 2023 08:53:08 +0530 Subject: [PATCH] Fixed syntax warning --- deepforest/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepforest/main.py b/deepforest/main.py index bf66f368..ae0fe2ca 100644 --- a/deepforest/main.py +++ b/deepforest/main.py @@ -107,7 +107,7 @@ def use_release(self, check_release=True): # Download latest model from github release release_tag, self.release_state_dict = utilities.use_release( check_release=check_release) - if self.config["architecture"] is not "retinanet": + if self.config["architecture"] != "retinanet": warnings.warn( "The config file specifies architecture {}, but the release model is torchvision retinanet. Reloading with deepforest.main with a retinanet model" .format(self.config["architecture"]))