Fix broadcast operators from #429 using deprecated syntax

mratsim · Apr 19, 2020 · bde1880 · bde1880
1 parent db3f637
commit bde1880
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/src/tensor/operators_comparison.nim b/src/tensor/operators_comparison.nim
@@ -98,38 +98,38 @@ template gen_broadcasted_scalar_comparison(op: untyped): untyped {.dirty.} =
   result = map_inline(t):
     op(x, value)
 
-proc `.==`*[T](t: Tensor[T], value : T): Tensor[bool] {.noInit.} =
+proc `==.`*[T](t: Tensor[T], value : T): Tensor[bool] {.noInit.} =
   ## Tensor element-wise equality with scalar
   ## Returns:
   ##   - A tensor of boolean
   gen_broadcasted_scalar_comparison(`==`)
 
 
-proc `.!=`*[T](t: Tensor[T], value : T): Tensor[bool] {.noInit.} =
+proc `!=.`*[T](t: Tensor[T], value : T): Tensor[bool] {.noInit.} =
   ## Tensor element-wise inequality with scalar
   ## Returns:
   ##   - A tensor of boolean
   gen_broadcasted_scalar_comparison(`!=`)
 
-proc `.<=`*[T](t: Tensor[T], value : T): Tensor[bool] {.noInit.} =
+proc `<=.`*[T](t: Tensor[T], value : T): Tensor[bool] {.noInit.} =
   ## Tensor element-wise lesser or equal with scalar
   ## Returns:
   ##   - A tensor of boolean
   gen_broadcasted_scalar_comparison(`<=`)
 
-proc `.<`*[T](t: Tensor[T], value : T): Tensor[bool] {.noInit.} =
+proc `<.`*[T](t: Tensor[T], value : T): Tensor[bool] {.noInit.} =
   ## Tensor element-wise lesser than a scalar
   ## Returns:
   ##   - A tensor of boolean
   gen_broadcasted_scalar_comparison(`<`)
 
-proc `.>=`*[T](t: Tensor[T], value : T): Tensor[bool] {.noInit.} =
+proc `>=.`*[T](t: Tensor[T], value : T): Tensor[bool] {.noInit.} =
   ## Tensor element-wise greater or equal than a scalar
   ## Returns:
   ##   - A tensor of boolean
   gen_broadcasted_scalar_comparison(`>=`)
 
-proc `.>`*[T](t: Tensor[T], value : T): Tensor[bool] {.noInit.} =
+proc `>.`*[T](t: Tensor[T], value : T): Tensor[bool] {.noInit.} =
   ## Tensor element-wise greater than a scalar
   ## Returns:
   ##   - A tensor of boolean

diff --git a/tests/tensor/test_selectors.nim b/tests/tensor/test_selectors.nim
@@ -143,7 +143,7 @@ suite "Selectors":
                [2, 2]].toTensor
 
       let rowsum = x.sum(axis = 1)
-      let cond = rowsum .<= 2
+      let cond = rowsum <=. 2
       let r = x.masked_axis_select(cond.squeeze(), axis = 0)
 
       let expected = [[0, 1],
@@ -179,7 +179,7 @@ suite "Selectors":
                [ 1,  2, 0],
                [ 1, -1, 1]].toTensor
 
-      let cond = squeeze(a.sum(axis = 0) .> 1)
+      let cond = squeeze(a.sum(axis = 0) >. 1)
       a.masked_axis_fill(cond, axis = 1, -10)
 
       let expected = [[-1, -2, -10],