Format.

Author: khatchad

com.ibm.wala.cast.python.ml/source/com/ibm/wala/cast/python/ml/client/PythonTensorAnalysisEngine.java

@@ -709,7 +709,8 @@ public TensorTypeAnalysis performAnalysis(PropagationCallGraphBuilder builder)
       init.put(e.getKey(), Set.of(e.getValue()));
 
     Map<PointsToSetVariable, TensorType> setCalls = HashMapFactory.make();
-    Map<PointsToSetVariable, TensorType> set_shapes = getShapeSourceCalls(set_shape, builder, 1); // TODO: What if you used tf.ones() here?
+    Map<PointsToSetVariable, TensorType> set_shapes =
+        getShapeSourceCalls(set_shape, builder, 1); // TODO: What if you used tf.ones() here?
 
     for (Map.Entry<PointsToSetVariable, TensorType> x : set_shapes.entrySet()) {
       LocalPointerKey localPointerKey = (LocalPointerKey) x.getKey().getPointerKey();
@@ -768,7 +769,8 @@ private Set<TensorType> getTensorType(
     TypeReference calledFunction = node.getMethod().getDeclaringClass().getReference();
     logger.info("Getting tensor type for call to: " + calledFunction.getName() + ".");
 
-    if (calledFunction.equals(ONES.getDeclaringClass())) { // TODO: This can also be a tuple of Tensor.
+    if (calledFunction.equals(
+        ONES.getDeclaringClass())) { // TODO: This can also be a tuple of Tensor.
       // This is a call to `ones()`. The shape is in the first explicit argument.
       PointerAnalysis<InstanceKey> pointerAnalysis = builder.getPointerAnalysis();
       // TODO: Handle keyword arguments.