Add review comments

apache · Dec 21, 2017 · 2df1e91 · 2df1e91
1 parent b7de0ec
commit 2df1e91
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdks/python/apache_beam/transforms/trigger.py b/sdks/python/apache_beam/transforms/trigger.py
@@ -378,8 +378,8 @@ class AfterCount(TriggerFn):
   COUNT_TAG = _CombiningValueStateTag('count', combiners.CountCombineFn())
 
   def __init__(self, count):
-    if count == 0:
-      raise ValueError("count (%s) must be a non-zero natural number." % count)
+    if not isinstance(count, int) or count < 1:
+      raise ValueError("count (%d) must be a non-zero natural number." % count)
     self.count = count
 
   def __repr__(self):