Skip to content

Commit fdb8738

Browse files
committed
Fix method name.
1 parent 7f62689 commit fdb8738

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

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

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -326,10 +326,7 @@ else if (valueIK instanceof AllocationSiteInNode) {
326326
* type literals.
327327
* @return A set of possible dtypes of the tensor returned by this generator.
328328
*/
329-
protected EnumSet<DType>
330-
getDTypesFromShapeArgument( // TODO: Shouldn't this be "fromDTypeArgument" or simply
331-
// "fromArgument"?
332-
PropagationCallGraphBuilder builder, Iterable<InstanceKey> pointsToSet) {
329+
protected EnumSet<DType> getDTypesFromDTypeArgument(PropagationCallGraphBuilder builder, Iterable<InstanceKey> pointsToSet) {
333330
EnumSet<DType> ret = EnumSet.noneOf(DType.class);
334331
PointerAnalysis<InstanceKey> pointerAnalysis = builder.getPointerAnalysis();
335332

@@ -452,7 +449,7 @@ protected EnumSet<DType> getDTypes(PropagationCallGraphBuilder builder) {
452449
if (pointsToSet == null || pointsToSet.isEmpty()) return getDefaultDTypes(builder);
453450
else
454451
// The dtype points-to set is non-empty, meaning that the dtype was explicitly set.
455-
return getDTypesFromShapeArgument(builder, pointsToSet);
452+
return getDTypesFromDTypeArgument(builder, pointsToSet);
456453
}
457454

458455
/**

0 commit comments

Comments
 (0)