Fixed syntax warning (#565)

weecology · Dec 6, 2023 · 64a7230 · 64a7230
1 parent 8c94681
commit 64a7230
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"]))