Fix invalid type annotation. (#48)

`ops` is a module not a type.

Author: hertschuh

keras_rs/src/utils/pairwise_loss_utils.py

@@ -6,7 +6,9 @@
 from keras_rs.src.utils.keras_utils import check_shapes_compatible
 
 
-def apply_pairwise_op(x: types.Tensor, op: ops) -> types.Tensor:
+def apply_pairwise_op(
+    x: types.Tensor, op: Callable[[types.Tensor, types.Tensor], types.Tensor]
+) -> types.Tensor:
     return op(
         ops.expand_dims(x, axis=-1),
         ops.expand_dims(x, axis=-2),