Update nsl/research/carls to no longer rely on keras 1D->2D data upranking. We are planning to remove this behavior from the keras codebase. We are fixing various broken tests to minimize the impact of this breaking change.

PiperOrigin-RevId: 385853308

Author: LukeWood

research/carls/graph_regularization.py

@@ -120,6 +120,11 @@ def call(self, inputs, training=False, **kwargs):
     # Invoke the call() function of the neighbor features layer directly instead
     # of invoking it as a callable to avoid Keras from wrapping placeholder
     # tensors with the tf.identity() op.
+
+    for feature_name, feature_value in inputs.items():
+      if len(feature_value.shape) < 2:
+        inputs[feature_name] = tf.expand_dims(feature_value, axis=-1)
+
     sample_features, nbr_features, nbr_weights = self.nbr_features_layer.call(
         inputs)
     base_output = self._infer_and_update(