From 693f4d4d68d97d42fc9e04eb43de5bb175dd76c1 Mon Sep 17 00:00:00 2001
From: Martin Reinecke <martin@mpa-garching.mpg.de>
Date: Tue, 18 Dec 2018 11:10:32 +0100
Subject: [PATCH] allow multiplying operators with scalars
---
nifty5/operators/operator.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/nifty5/operators/operator.py b/nifty5/operators/operator.py
index 4a11dfb50..ed65befab 100644
--- a/nifty5/operators/operator.py
+++ b/nifty5/operators/operator.py
@@ -59,9 +59,11 @@ class Operator(NiftyMetaBase()):
return _OpChain.make((self, x))
def __mul__(self, x):
- if not isinstance(x, Operator):
- return NotImplemented
- return _OpProd(self, x)
+ if isinstance(x, Operator):
+ return _OpProd(self, x)
+ if np.isscalar(x):
+ return self.scale(x)
+ return NotImplemented
def __add__(self, x):
if not isinstance(x, Operator):
--
GitLab