From 1ff432327d12a14d131ee671fe1e18a2f209701e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Beno=C3=AEt=20Sierro?= <benoitsierro@bluewin.ch>
Date: Tue, 21 Dec 2021 11:53:32 +0100
Subject: [PATCH] working without phase effect

---
 src/scgenerator/const.py     | 2 ++
 src/scgenerator/operators.py | 2 +-
 src/scgenerator/plotting.py  | 1 -
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/scgenerator/const.py b/src/scgenerator/const.py
index 45df234..92610e3 100644
--- a/src/scgenerator/const.py
+++ b/src/scgenerator/const.py
@@ -81,6 +81,8 @@ VALID_VARIABLE = {
     "num",
 }
 
+INVALID_VARIABLE = {"repeat"}
+
 
 MANDATORY_PARAMETERS = [
     "name",
diff --git a/src/scgenerator/operators.py b/src/scgenerator/operators.py
index 1b13bfb..f773c40 100644
--- a/src/scgenerator/operators.py
+++ b/src/scgenerator/operators.py
@@ -831,7 +831,7 @@ class Plasma(Operator):
         return dict(ionization_fraction=self.ionization_fraction)
 
 
-class NoPlasma(NoOpTime, Plasma):
+class NoPlasma(NoOpFreq, Plasma):
     pass
 
 
diff --git a/src/scgenerator/plotting.py b/src/scgenerator/plotting.py
index b2be66c..d906e3c 100644
--- a/src/scgenerator/plotting.py
+++ b/src/scgenerator/plotting.py
@@ -453,7 +453,6 @@ def transform_2D_propagation(
         raise ValueError(f"shape was {values.shape}. Can only plot 2D array")
     is_complex, x_axis, plt_range = prep_plot_axis(values, plt_range, params)
     if is_complex or any(values.ravel() < 0):
-        print("squared")
         values = abs2(values)
     # if params.full_field and plt_range.unit.type == "TIME":
     #     values = envelope_2d(x_axis, values)