Changed function signatures to comply with updated tensorflow require…

…ments. -Subramanian Iyer <subramanian.iyer@paramount.com>
recommenders-team · Aug 15, 2024 · 551ec31 · 551ec31
1 parent 0296e95
commit 551ec31
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/recommenders/models/sasrec/ssept.py b/recommenders/models/sasrec/ssept.py
@@ -122,7 +122,7 @@ def call(self, x, training):
         # --- ATTENTION BLOCKS ---
         seq_attention = seq_embeddings  # (b, s, h1 + h2)
 
-        seq_attention = self.encoder(seq_attention, training, mask)
+        seq_attention = self.encoder(seq_attention, training=training, mask=mask)
         seq_attention = self.layer_normalization(seq_attention)  # (b, s, h1+h2)
 
         # --- PREDICTION LAYER ---
@@ -197,7 +197,7 @@ def predict(self, inputs):
 
         seq_embeddings *= mask
         seq_attention = seq_embeddings
-        seq_attention = self.encoder(seq_attention, training, mask)
+        seq_attention = self.encoder(seq_attention, training=training, mask=mask)
         seq_attention = self.layer_normalization(seq_attention)  # (b, s, h1+h2)
         seq_emb = tf.reshape(
             seq_attention,