keras-rs-nightly 0.0.1.dev2025040803__py3-none-any.whl → 0.0.1.dev2025040903__py3-none-any.whl

keras_rs/src/layers/retrieval/remove_accidental_hits.py

@@ -19,8 +19,8 @@ class RemoveAccidentalHits(keras.layers.Layer):
 
     def call(
         self,
-        labels: types.Tensor,
         logits: types.Tensor,
+        labels: types.Tensor,
         candidate_ids: types.Tensor,
     ) -> types.Tensor:
         """Zeroes selected logits.
@@ -29,16 +29,16 @@ class RemoveAccidentalHits(keras.layers.Layer):
         have the same ID as the positive candidate in that row.
 
         Args:
-          labels: one-hot labels tensor, typically
-              `[batch_size, num_candidates]` but can have more dimensions or be
-              1D as `[num_candidates]`.
-          logits: logits tensor. Must have the same shape as `labels`.
-          candidate_ids: candidate identifiers tensor, can be `[num_candidates]`
-              or `[batch_size, num_candidates]` or have more dimensions as long
-              as they match the last dimensions of `labels`.
+            logits: logits tensor, typically `[batch_size, num_candidates]` but
+                can have more dimensions or be 1D as `[num_candidates]`.
+            labels: one-hot labels tensor, must be the same shape as `logits`.
+            candidate_ids: candidate identifiers tensor, can be
+                `[num_candidates]` or `[batch_size, num_candidates]` or have
+                more dimensions as long as they match the last dimensions of
+                `labels`.
 
         Returns:
-          logits: Modified logits.
+            logits: Modified logits with the same shape as the input logits.
         """
         # A more principled way is to implement
         # `softmax_cross_entropy_with_logits` with a input mask. Here we

keras_rs/src/version.py

@@ -1,7 +1,7 @@
 from keras_rs.src.api_export import keras_rs_export
 
 # Unique source of truth for the version number.
-__version__ = "0.0.1.dev2025040803"
+__version__ = "0.0.1.dev2025040903"
 
 
 @keras_rs_export("keras_rs.version")

{keras_rs_nightly-0.0.1.dev2025040803.dist-info → keras_rs_nightly-0.0.1.dev2025040903.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: keras-rs-nightly
-Version: 0.0.1.dev2025040803
+Version: 0.0.1.dev2025040903
 Summary: Multi-backend recommender systems with Keras 3.
 Author-email: Keras RS team <keras-rs@google.com>
 License: Apache License 2.0

{keras_rs_nightly-0.0.1.dev2025040803.dist-info → keras_rs_nightly-0.0.1.dev2025040903.dist-info}/RECORD

@@ -5,7 +5,7 @@ keras_rs/api/losses/__init__.py,sha256=LGW7eHQh8FbQXdMV1s9zJpbloVlz_Zlo51sorWAvF
 keras_rs/src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_rs/src/api_export.py,sha256=RsmG-DvO-cdFeAF9W6LRzms0kvtm-Yp9BAA_d-952zI,510
 keras_rs/src/types.py,sha256=UyOdgjqrqg_b58opnY8n6gTiDHKVR8z_bmEruehERBk,514
-keras_rs/src/version.py,sha256=a9BItWgRhEYrbx2zCLxStYnDVn5qAEy8Ovsszh2TS_o,222
+keras_rs/src/version.py,sha256=wVn06NpH2Q2aM6jiAg5DElYd-lEp6sbrNZSQL-1PPss,222
 keras_rs/src/layers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_rs/src/layers/feature_interaction/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_rs/src/layers/feature_interaction/dot_interaction.py,sha256=jGHcg0EiWxth6LTxG2yWgHcyx_GXrxvA61uQqpPfnDQ,6900
@@ -13,7 +13,7 @@ keras_rs/src/layers/feature_interaction/feature_cross.py,sha256=5OCSI0vFYzJNmgkK
 keras_rs/src/layers/retrieval/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_rs/src/layers/retrieval/brute_force_retrieval.py,sha256=mohILOt6PC6jHBztaowDbj3QBnSetuvkq55FmE39PlY,7321
 keras_rs/src/layers/retrieval/hard_negative_mining.py,sha256=CY8-3W52ZBIFcEfvjXJxbFltD6ulXl4-sZCRF6stIEc,4119
-keras_rs/src/layers/retrieval/remove_accidental_hits.py,sha256=fiFQLlkMBXhG8V7a8mv_hKOwlqEJeUiMBYUVQw1woTE,3270
+keras_rs/src/layers/retrieval/remove_accidental_hits.py,sha256=Z84z2YgKspKeNdc5id8lf9TAyFsbCCz3acJxiKXYipc,3324
 keras_rs/src/layers/retrieval/sampling_probability_correction.py,sha256=80vgOPfBiF-PC0dSyqS57IcIxOxi_Q_R7eSXHn1G0yI,1437
 keras_rs/src/losses/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_rs/src/losses/pairwise_hinge_loss.py,sha256=vqDGd-OnZxiqdeE6vuabE8BKDfill3D2GM0lW5JUmsg,922
@@ -24,7 +24,7 @@ keras_rs/src/losses/pairwise_soft_zero_one_loss.py,sha256=XBej5nybFXEQ-Vp6GLvNmq
 keras_rs/src/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras_rs/src/utils/keras_utils.py,sha256=IjWSRieBkv7UX12qgUoI1tcOeISstCLRSTqSHpT06yE,1275
 keras_rs/src/utils/pairwise_loss_utils.py,sha256=5SAqA3z30A1awzV9l5oVbcno5Z6HXARkNcUFTPL7_jg,3380
-keras_rs_nightly-0.0.1.dev2025040803.dist-info/METADATA,sha256=oC2tgoUja1GWeMPNM6ZFmrBJNnuPof-Eb8LioVF3kYs,3547
-keras_rs_nightly-0.0.1.dev2025040803.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-keras_rs_nightly-0.0.1.dev2025040803.dist-info/top_level.txt,sha256=pWs8X78Z0cn6lfcIb9VYOW5UeJ-TpoaO9dByzo7_FFo,9
-keras_rs_nightly-0.0.1.dev2025040803.dist-info/RECORD,,
+keras_rs_nightly-0.0.1.dev2025040903.dist-info/METADATA,sha256=KWzgc2oFlkqoj-qkXObJvbUissueApGcNWH9IZpCQWw,3547
+keras_rs_nightly-0.0.1.dev2025040903.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+keras_rs_nightly-0.0.1.dev2025040903.dist-info/top_level.txt,sha256=pWs8X78Z0cn6lfcIb9VYOW5UeJ-TpoaO9dByzo7_FFo,9
+keras_rs_nightly-0.0.1.dev2025040903.dist-info/RECORD,,