Mark PartialGroupByKeyOperation can restart.

[] ------------- Created by MOE: http://code.google.com/p/moe-java MOE_MIGRATED_REVID=87206956
ngrunwald · Feb 27, 2015 · b76455f · b76455f
1 parent 48dfb0c
commit b76455f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/...ain/java/com/google/cloud/dataflow/sdk/util/common/worker/PartialGroupByKeyOperation.java b/...ain/java/com/google/cloud/dataflow/sdk/util/common/worker/PartialGroupByKeyOperation.java
@@ -194,6 +194,12 @@ public void finish() throws Exception {
     }
   }
 
+  @Override
+  public boolean supportsRestart() {
+    // SizeEstimators are safe to be reused.
+    return true;
+  }
+
   /**
    * Sets the maximum amount of memory the grouping table is allowed to
    * consume before it has to be flushed.