diff --git a/.github/actionlint.yaml b/.github/actionlint.yaml
index e0fa30b4bf..f5f4e4c59c 100644
--- a/.github/actionlint.yaml
+++ b/.github/actionlint.yaml
@@ -3,3 +3,4 @@ self-hosted-runner:
   labels:
     - 1GPU
     - 2GPU
+    - linux-amd64-gpu-p100-latest-1
diff --git a/merlin/models/tf/models/base.py b/merlin/models/tf/models/base.py
index ec1f950e77..af0cb14c7e 100644
--- a/merlin/models/tf/models/base.py
+++ b/merlin/models/tf/models/base.py
@@ -1798,7 +1798,7 @@ def build(self, input_shape=None):
 
         self.built = True
 
-    def call(self, inputs, targets=None, training=False, testing=False, output_context=False):
+    def call(self, inputs, targets=None, training=None, testing=None, output_context=None):
         """
         Method for forward pass of the model.
 
@@ -1820,6 +1820,9 @@ def call(self, inputs, targets=None, training=False, testing=False, output_conte
         Tensor or tuple of Tensor and ModelContext
             Output of the model, and optionally the context
         """
+        training = training or False
+        testing = testing or False
+        output_context = output_context or False
         outputs = inputs
         features = self._prepare_features(inputs, targets=targets)
         if isinstance(features, tuple):