Skip to content

Commit b3f9437

Browse files
committed
Add logging.
1 parent 69a0a89 commit b3f9437

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

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

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -736,7 +736,11 @@ private Set<TensorType> getTensorTypes(
736736
logger.info("Getting tensor types for source: " + source + ".");
737737

738738
TensorGenerator generator = TensorGeneratorFactory.getGenerator(source);
739-
return generator.getTensorTypes(builder);
739+
740+
Set<TensorType> tensorTypes = generator.getTensorTypes(builder);
741+
logger.info(() -> "Tensor types for source: " + source + " are: " + tensorTypes + ".");
742+
743+
return tensorTypes;
740744
}
741745

742746
private Map<PointsToSetVariable, TensorType> handleShapeSourceOp(

0 commit comments

Comments
 (0)