remove debug options for faster compile

Author: Cara Giovanetti

ABCMB/main.py

@@ -194,7 +194,7 @@ def run_cosmology_abbr(self, params : dict):
         """
 
         # let the user know the code is compiling
-        print("\n")
+        print("")
         print("            __")
         print("           /  \\")
         print("          /    \\")
@@ -207,7 +207,7 @@ def run_cosmology_abbr(self, params : dict):
         print("______/"+'\033[1m' +"/ /        \\ \\| _ \\\\ \\___ ||\\/||| - )"+"\033[0m"+"/\\  ")
         print("      "+'\033[1m' +"/_/          \\_\\___/ \\____|||  |||_-_)"+"\033[0m"+"  \\/\\ is compiling...")
         print("                                                 \\/\\")
-        print("\n")
+        print("")
 
         PT, BG = self.get_PTBG(params)
         output = ()

time_tests.py

@@ -1,60 +1,46 @@
-from classy import Class
-
 import sys
 sys.path.append('../')
-# sys.path.append('../JaxCMB')
-# print(sys.path)
-
 import os
-# os.environ.setdefault("JAX_PLATFORM_NAME", "cpu")
-# print(os.getcwd())
-
-import sys
 
-# assert "jax" not in sys.modules, "jax already imported: you must restart your runtime"
-# os.environ['XLA_FLAGS'] = "--xla_force_host_platform_device_count=8"
 
 import jax
 print(jax.devices())
-# import jax
 jax.config.update("jax_enable_x64", True)
-jax.config.update("jax_debug_nans", True)
 from ABCMB.main import Model
 import ABCMB.spectrum as spectrum
 from scipy.interpolate import interp1d
 import jax.numpy as jnp
 import numpy as np
 import matplotlib.pyplot as plt
-import pytest
-import numpy as np
-np.seterr(all='raise') 
+
 
 import time
 
 h = 0.6762
 
+model = Model(bbn_type='LINX')
+# model = Model()
+
 for i in range(2):
     start=time.time()
     # ABCMB:
-    params = {
-        'h': jnp.asarray(h),
-        'omega_cdm': jnp.asarray(0.1193),
-        'omega_b': jnp.asarray(0.0225),
-        'A_s': jnp.asarray(2.12424e-9),
-        'n_s': jnp.asarray(0.9709),
-        #'Neff': 3.044,
-        #'Delta_Neff_init': jnp.asarray(0.),
-        #'YHe': jnp.asarray(0.245),
-        'TCMB0': jnp.asarray(2.34865418e-4),
-        #'T_nu': jnp.asarray((4. / 11.)**(1. / 3.) * 2.34865418e-4),
-        'N_ncdm': jnp.asarray(0.),
-        'T_ncdm': jnp.asarray(0.71611),
-        'm_ncdm': jnp.asarray(0.06)
-    }
-
-    model = Model(bbn_type='LINX')
-    #model = Model()
-
+    # params = {
+    #     'h': jnp.asarray(h),
+    #     'omega_cdm': jnp.asarray(0.1193),
+    #     'omega_b': jnp.asarray(0.0225),
+    #     'A_s': jnp.asarray(2.12424e-9),
+    #     'n_s': jnp.asarray(0.9709),
+    #     #'Neff': 3.044,
+    #     #'Delta_Neff_init': jnp.asarray(0.),
+    #     #'YHe': jnp.asarray(0.245),
+    #     'TCMB0': jnp.asarray(2.34865418e-4),
+    #     #'T_nu': jnp.asarray((4. / 11.)**(1. / 3.) * 2.34865418e-4),
+    #     # 'N_ncdm': jnp.asarray(0.),
+    #     # 'T_ncdm': jnp.asarray(0.71611),
+    #     # 'm_ncdm': jnp.asarray(0.06)
+    # }
+    params = {}
+    
     ell, ABC_Cl = model.run_cosmology(params)
 
     print(ABC_Cl[0])