tf-keras-nightly 2.20.0.dev2025032009__py3-none-any.whl → 2.20.0.dev2025032109__py3-none-any.whl

tf_keras/__init__.py

@@ -27,4 +27,4 @@ from tf_keras.src.engine.sequential import Sequential
 from tf_keras.src.engine.training import Model
 
 
-__version__ = "2.20.0.dev2025032009"
+__version__ = "2.20.0.dev2025032109"

tf_keras/src/saving/saving_api.py

@@ -24,6 +24,7 @@ from tensorflow.python.util.tf_export import keras_export
 
 from tf_keras.src.saving import saving_lib
 from tf_keras.src.saving.legacy import save as legacy_sm_saving_lib
+from tf_keras.src.saving.legacy import saving_utils
 from tf_keras.src.utils import io_utils
 
 try:
@@ -75,8 +76,7 @@ class SupportWriteToRemote:
     supports remoted saved model out of the box.
     """
 
-    def __init__(self, filepath, overwrite=True, save_format=None):
-        save_format = get_save_format(filepath, save_format=save_format)
+    def __init__(self, filepath, overwrite, save_format):
         self.overwrite = overwrite
         if saving_lib.is_remote_path(filepath) and save_format != "tf":
             self.temp_directory = tempfile.TemporaryDirectory()
@@ -191,14 +191,14 @@ def save_model(model, filepath, overwrite=True, save_format=None, **kwargs):
     when loading the model. See the `custom_objects` argument in
     `tf.keras.saving.load_model`.
     """
+    save_format = get_save_format(filepath, save_format)
+
     # Supports remote paths via a temporary file
     with SupportWriteToRemote(
         filepath,
         overwrite=overwrite,
         save_format=save_format,
     ) as local_filepath:
-        save_format = get_save_format(filepath, save_format)
-
         # Deprecation warnings
         if save_format == "h5":
             warnings.warn(
@@ -307,8 +307,12 @@ def load_model(
 
 
 def save_weights(model, filepath, overwrite=True, **kwargs):
+    save_format = get_save_weights_format(filepath)
+
     # Supports remote paths via a temporary file
-    with SupportWriteToRemote(filepath, overwrite=overwrite) as local_filepath:
+    with SupportWriteToRemote(
+        filepath, overwrite=overwrite, save_format=save_format
+    ) as local_filepath:
         if str(local_filepath).endswith(".weights.h5"):
             # If file exists and should not be overwritten.
             try:
@@ -386,3 +390,12 @@ def get_save_format(filepath, save_format):
     else:
         return "h5"
 
+
+def get_save_weights_format(filepath):
+    filepath = io_utils.path_to_string(filepath)
+    filepath_is_h5 = saving_utils.is_hdf5_filepath(filepath)
+    if filepath_is_h5:
+        return "h5"
+    else:
+        return "tf"
+

{tf_keras_nightly-2.20.0.dev2025032009.dist-info → tf_keras_nightly-2.20.0.dev2025032109.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tf_keras-nightly
-Version: 2.20.0.dev2025032009
+Version: 2.20.0.dev2025032109
 Summary: Deep learning for humans.
 Home-page: https://keras.io/
 Download-URL: https://github.com/keras-team/tf-keras/tags

{tf_keras_nightly-2.20.0.dev2025032009.dist-info → tf_keras_nightly-2.20.0.dev2025032109.dist-info}/RECORD

@@ -1,4 +1,4 @@
-tf_keras/__init__.py,sha256=eIZSMQ5R2EgQGr1LYuqqY0omLphlSiVR9-9UW_tGjIU,911
+tf_keras/__init__.py,sha256=xXFhiHhs0Iue2KnVgnIGN4N7gFb7t9KxLAG_uPhzzT4,911
 tf_keras/__internal__/__init__.py,sha256=OHQbeIC0QtRBI7dgXaJaVbH8F00x8dCI-DvEcIfyMsE,671
 tf_keras/__internal__/backend/__init__.py,sha256=LnMs2A6685gDG79fxqmdulIYlVE_3WmXlBTBo9ZWYcw,162
 tf_keras/__internal__/layers/__init__.py,sha256=F5SGMhOTPzm-PR44VrfinURHcVeQPIEdwnZlAkSTB3A,176
@@ -537,7 +537,7 @@ tf_keras/src/preprocessing/text.py,sha256=aomzwE3G2ErwzgL_Dj3ERA_2k7TZclaPMTBTrW
 tf_keras/src/saving/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 tf_keras/src/saving/object_registration.py,sha256=N8aV6eqREYjW2ueQpL3guYHyh5KXuun3DZAlmjfYrTA,7830
 tf_keras/src/saving/pickle_utils.py,sha256=5GtHzwNWVaYfZ-0zn69-zn2yv3R6JUwzHOOamnjP7r0,2605
-tf_keras/src/saving/saving_api.py,sha256=q2_-CsVg81HeHnXDkRqpoBwI3Mpp_gQXjBidr5V_RLo,14838
+tf_keras/src/saving/saving_api.py,sha256=7uIWY4uwQMtETP5yRjZYPii-E-sWzSuK0ljVMfnZe_k,15119
 tf_keras/src/saving/saving_lib.py,sha256=Rk5rOvxEmCvwUlG3bS0QpOgtURayYmIAZbh2GeTuUOc,24272
 tf_keras/src/saving/serialization_lib.py,sha256=kX4qf_fRp4LySkH9FU37DMd0AXxiUrXKT-VLR3JPl7w,30152
 tf_keras/src/saving/legacy/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
@@ -606,7 +606,7 @@ tf_keras/src/utils/legacy/__init__.py,sha256=EfMmeHYDzwvxNaktPhQbkTdcPSIGCqMhBND
 tf_keras/utils/__init__.py,sha256=b7_d-USe_EmLo02_P99Q1rUCzKBYayPCfiYFStP-0nw,2735
 tf_keras/utils/experimental/__init__.py,sha256=DzGogE2AosjxOVILQBT8PDDcqbWTc0wWnZRobCdpcec,97
 tf_keras/utils/legacy/__init__.py,sha256=7ujlDa5HeSRcth2NdqA0S1P2-VZF1kB3n68jye6Dj-8,189
-tf_keras_nightly-2.20.0.dev2025032009.dist-info/METADATA,sha256=i85ekqBcTzRZ-cqN3mnctx96L4loEGxXarb9Afhl070,1857
-tf_keras_nightly-2.20.0.dev2025032009.dist-info/WHEEL,sha256=tTnHoFhvKQHCh4jz3yCn0WPTYIy7wXx3CJtJ7SJGV7c,91
-tf_keras_nightly-2.20.0.dev2025032009.dist-info/top_level.txt,sha256=LC8FK7zHDNKxB17C6lGKvrZ_fZZGJsRiBK23SfiDegY,9
-tf_keras_nightly-2.20.0.dev2025032009.dist-info/RECORD,,
+tf_keras_nightly-2.20.0.dev2025032109.dist-info/METADATA,sha256=7_IStzcQqT3ZEgiYsF3ZEZIhNwAOdkpJGdsZBQ3l6WM,1857
+tf_keras_nightly-2.20.0.dev2025032109.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
+tf_keras_nightly-2.20.0.dev2025032109.dist-info/top_level.txt,sha256=LC8FK7zHDNKxB17C6lGKvrZ_fZZGJsRiBK23SfiDegY,9
+tf_keras_nightly-2.20.0.dev2025032109.dist-info/RECORD,,

{tf_keras_nightly-2.20.0.dev2025032009.dist-info → tf_keras_nightly-2.20.0.dev2025032109.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (77.0.1)
+Generator: setuptools (77.0.3)
 Root-Is-Purelib: true
 Tag: py3-none-any