minor: added __call__ to TorchOperator

mrava87 · mrava87 · commit fee227f34542 · 2023-02-27T21:58:57.000+03:00
diff --git a/pylops/torchoperator.py b/pylops/torchoperator.py
@@ -82,6 +82,9 @@ def __init__(
             )
         self.Top = _TorchOperator.apply
 
+    def __call__(self, x):
+        return self.apply(x)
+
     def apply(self, x: TensorTypeLike) -> TensorTypeLike:
         """Apply forward pass to input vector
 

Original file line number	Diff line number	Diff line change
`@@ -82,6 +82,9 @@ def __init__(`
`82`	`82`	`)`
`83`	`83`	`self.Top = _TorchOperator.apply`
`84`	`84`
	`85`	`+ def __call__(self, x):`
	`86`	`+ return self.apply(x)`
	`87`	`+`
`85`	`88`	`def apply(self, x: TensorTypeLike) -> TensorTypeLike:`
`86`	`89`	`"""Apply forward pass to input vector`
`87`	`90`