update post_sigma_OHE.py

quanshengwu · May 7, 2024 · 9fdb898 · 9fdb898
1 parent d807821
commit 9fdb898
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/examples/Cu/post_sigma_OHE.py b/examples/Cu/post_sigma_OHE.py
@@ -5,7 +5,7 @@
 # ( in testing )
 
 '''
-0. Required Packages: numpy, matplotlib, scipy, re, os
+0. Required Packages: PYTHON3, numpy, matplotlib, scipy, re, os
    Required files : wt.in, sigma_bands_mu_*.dat
    Copy this file to your work folder where contains wt.in and sigma files 
    Usage : python post_sigma_OHE.py
@@ -83,7 +83,10 @@ def readwtin():
     OmegaMax = parameters['OMEGAMAX']
     OmegaNum = parameters['OMEGANUM']
     BTauMax = parameters['BTAUMAX']
-    BTauNum = parameters['BTAUNUM']
+    try:
+        BTauNum = parameters['BTAUNUM']
+    except:
+        BTauNum = parameters['NBTAU']
 
     print('Tmin = ',Tmin, '\nTmax = ',Tmax, '\nNumT = ',NumT, '\nOmegaMin = ',OmegaMin, '\nOmegaMax = ',
             OmegaMax, '\nOmegaNum = ',OmegaNum, '\nBTauMax = ',BTauMax, '\nBTauNum = ',BTauNum)

diff --git a/useful_scripts/post_sigma_OHE/post_sigma_OHE.py b/useful_scripts/post_sigma_OHE/post_sigma_OHE.py
@@ -5,7 +5,7 @@
 # ( in testing )
 
 '''
-0. Required Packages: numpy, matplotlib, scipy, re, os
+0. Required Packages: PYTHON3, numpy, matplotlib, scipy, re, os
    Required files : wt.in, sigma_bands_mu_*.dat
    Copy this file to your work folder where contains wt.in and sigma files 
    Usage : python post_sigma_OHE.py
@@ -85,7 +85,7 @@ def readwtin():
     BTauMax = parameters['BTAUMAX']
     try:
         BTauNum = parameters['BTAUNUM']
-    except
+    except:
         BTauNum = parameters['NBTAU']
 
     print('Tmin = ',Tmin, '\nTmax = ',Tmax, '\nNumT = ',NumT, '\nOmegaMin = ',OmegaMin, '\nOmegaMax = ',