From 3911af0c2648b3b90de050b1402ba371beb353fe Mon Sep 17 00:00:00 2001
From: Cammille Carinan <cammille.carinan@xfel.eu>
Date: Tue, 25 May 2021 18:11:45 +0200
Subject: [PATCH] Add __all__ and adapt __init__ files
---
src/toolbox_scs/base/__init__.py | 5 +++++
src/toolbox_scs/base/knife_edge.py | 5 +++--
2 files changed, 8 insertions(+), 2 deletions(-)
diff --git a/src/toolbox_scs/base/__init__.py b/src/toolbox_scs/base/__init__.py
index e69de29..367a07d 100644
--- a/src/toolbox_scs/base/__init__.py
+++ b/src/toolbox_scs/base/__init__.py
@@ -0,0 +1,5 @@
+from . import knife_edge as knife_edge_module
+from .knife_edge import *
+
+
+__all__ = knife_edge_module.__all__
diff --git a/src/toolbox_scs/base/knife_edge.py b/src/toolbox_scs/base/knife_edge.py
index adabfd4..7e9fbca 100644
--- a/src/toolbox_scs/base/knife_edge.py
+++ b/src/toolbox_scs/base/knife_edge.py
@@ -3,10 +3,11 @@ from scipy import special
from scipy.optimize import curve_fit
+__all__ = ['knife_edge', 'knife_edge_base']
+
+
def knife_edge(positions, intensities, axisRange=None, p0=None):
"""
- The base implementation of the knife-edge scan analysis.
-
Calculates the beam radius at 1/e^2 from a knife-edge scan by
fitting with erfc function: f(a,b,u) = a*erfc(u) + b or
where u = sqrt(2)*(x-x0)/w0 with w0 the beam radius at 1/e^2
--
GitLab