v__0.185

math-econ-code · Dec 7, 2024 · 8bdfb7d · 8bdfb7d
1 parent b6d3252
commit 8bdfb7d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/mec/blp.py b/mec/blp.py
@@ -59,11 +59,11 @@ def pi_invs(pi_t_y,theLambda_k_l,eta_t_i_k, xi_l_y,maxit = 100000, reltol=1E-8,
 
 
 def pi_inv(pi_y,theLambda_k_l,eta_i_k, xi_l_y,maxit = 100000, reltol=1E-8, require_der = 0 ):
-    U_t_y, dUdLambda_t_y_k_l = pi_invs(pi_y[None,:],theLambda_k_l,eta_i_k[None,:,:], xi_l_y,maxit , reltol, require_der )
+    res = pi_invs(pi_y[None,:],theLambda_k_l,eta_i_k[None,:,:], xi_l_y,maxit , reltol, require_der )
     if require_der>0:
-        return U_t_y.squeeze(axis=0), dUdLambda_t_y_k_l.squeeze(axis=0)
+        return [res[0].squeeze(axis=0), res[1].squeeze(axis=0)]
     else:
-        return U_t_y.squeeze(axis=0), None
+        return [res[0].squeeze(axis=0) ]
 
 
 

diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
 	name="mec",
-	version="0.184",
+	version="0.185",
 	authors=["Alfred Galichon"],
 	author_email="ag133@nyu.edu",
 	licence="",