reformat

Author: leondgarse

keras_cv_attention_models/backend.py

@@ -29,7 +29,10 @@
 
         is_tf_use_legacy_keras = os.environ.get("TF_USE_LEGACY_KERAS", "0") == "1"
         if int(tf_version[0]) >= 2 and int(tf_version[1]) >= 16 and int(tf_keras_version[0]) >= 3:
-            print("[WARNING] Currently tensorflow>=2.16 with keras 3 not supported, try pip install tf-keras~=2.16 and set TF_USE_LEGACY_KERAS=1 if error.")
+            suggest = ".".join(tf_version[:2])
+            print(
+                f"[WARNING] Currently tensorflow>=2.16 with keras 3 not supported, try pip install tf-keras~={suggest} and set TF_USE_LEGACY_KERAS=1 if error."
+            )
         elif int(tf_version[0]) >= 2 and int(tf_version[1]) >= 16 and not is_tf_use_legacy_keras:
             os.environ["TF_USE_LEGACY_KERAS"] = "1"
             print("[WARNING] Setting TF_USE_LEGACY_KERAS=1. Make sure this is ahead of importing tensorflow or keras.")

keras_cv_attention_models/clip/tokenizer.py

@@ -1,4 +1,4 @@
-""" CLIP tokenizer
+"""CLIP tokenizer
 
 Copied from https://github.com/mlfoundations/open_clip/blob/main/src/open_clip/tokenizer.py
 Copied from https://github.com/openai/CLIP. Originally MIT License, Copyright (c) 2021 OpenAI.

keras_cv_attention_models/download_and_load.py

@@ -54,8 +54,8 @@ def reload_model_weights(
         file_name = os.path.basename(url)
         try:
             pretrained_model = backend.get_file(file_name, url, cache_subdir="models", file_hash=cur_file_hash)
-        except:
-            print("[Error] will not load weights, url not found or download failed:", url)
+        except Exception as ee:
+            print(f"[Error] will not load weights, url not found or download failed. Error message: {ee}, url: {url}")
             return None
         else:
             print(">>>> Load pretrained from:", pretrained_model)

keras_cv_attention_models/efficientnet/efficientnet_edgetpu.py

@@ -1,4 +1,4 @@
-""" Creates an EfficientNet-EdgeTPU model
+"""Creates an EfficientNet-EdgeTPU model
 Ref impl: https://github.com/tensorflow/tpu/tree/master/models/official/efficientnet/edgetpu
 """
 

keras_cv_attention_models/version.py

@@ -1 +1 @@
-__version__ = "1.4.2"
+__version__ = "1.4.3"

setup.py

@@ -1,5 +1,4 @@
-""" Setup
-"""
+"""Setup"""
 
 from setuptools import setup, find_packages
 from codecs import open

setup_kecam.py

@@ -1,5 +1,4 @@
-""" Setup
-"""
+"""Setup"""
 
 from setuptools import setup, find_packages
 from codecs import open