Skip to content

Commit 5a5f1fd

Browse files
committed
Fix version query for reduction ops tests.
1 parent 74f8bff commit 5a5f1fd

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

tests/ops.py

+7-7
Original file line numberDiff line numberDiff line change
@@ -578,7 +578,7 @@ def test_Sum(self):
578578
self.check(t)
579579
t = tf.reduce_sum(self.random(3, 4, 5), reduction_indices=0, keep_dims=True)
580580
self.check(t)
581-
if td._tf_version[:3] < (0, 12, 0):
581+
if td._tf_version[:3] >= (0, 12, 0):
582582
t = tf.reduce_sum(self.random(3, 4, 5), axis=[0, 1], keep_dims=True)
583583
self.check(t)
584584
t = tf.reduce_sum(self.random(3, 4, 5), axis=(0, 1), keep_dims=True)
@@ -589,42 +589,42 @@ def test_Sum(self):
589589
def test_Prod(self):
590590
t = tf.reduce_prod(self.random(3, 4, 5), reduction_indices=[0, 1], keep_dims=True)
591591
self.check(t)
592-
if td._tf_version[:3] < (0, 12, 0):
592+
if td._tf_version[:3] >= (0, 12, 0):
593593
t = tf.reduce_prod(self.random(3, 4, 5), axis=[0, 1], keep_dims=True)
594594
self.check(t)
595595

596596
def test_Min(self):
597597
t = tf.reduce_min(self.random(3, 4, 5), reduction_indices=[0, 1], keep_dims=True)
598598
self.check(t)
599-
if td._tf_version[:3] < (0, 12, 0):
599+
if td._tf_version[:3] >= (0, 12, 0):
600600
t = tf.reduce_min(self.random(3, 4, 5), axis=[0, 1], keep_dims=True)
601601
self.check(t)
602602

603603
def test_Max(self):
604604
t = tf.reduce_max(self.random(3, 4, 5), reduction_indices=[0, 1], keep_dims=True)
605605
self.check(t)
606-
if td._tf_version[:3] < (0, 12, 0):
606+
if td._tf_version[:3] >= (0, 12, 0):
607607
t = tf.reduce_max(self.random(3, 4, 5), axis=[0, 1], keep_dims=True)
608608
self.check(t)
609609

610610
def test_Mean(self):
611611
t = tf.reduce_mean(self.random(3, 4, 5), reduction_indices=[0, 1], keep_dims=True)
612612
self.check(t)
613-
if td._tf_version[:3] < (0, 12, 0):
613+
if td._tf_version[:3] >= (0, 12, 0):
614614
t = tf.reduce_mean(self.random(3, 4, 5), axis=[0, 1], keep_dims=True)
615615
self.check(t)
616616

617617
def test_All(self):
618618
t = tf.reduce_all(self.random(3, 4, 5), reduction_indices=[0, 1], keep_dims=True)
619619
self.check(t)
620-
if td._tf_version[:3] < (0, 12, 0):
620+
if td._tf_version[:3] >= (0, 12, 0):
621621
t = tf.reduce_all(self.random(3, 4, 5), axis=[0, 1], keep_dims=True)
622622
self.check(t)
623623

624624
def test_Any(self):
625625
t = tf.reduce_any(self.random(3, 4, 5), reduction_indices=[0, 1], keep_dims=True)
626626
self.check(t)
627-
if td._tf_version[:3] < (0, 12, 0):
627+
if td._tf_version[:3] >= (0, 12, 0):
628628
t = tf.reduce_any(self.random(3, 4, 5), axis=[0, 1], keep_dims=True)
629629
self.check(t)
630630

0 commit comments

Comments
 (0)