freealg 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

freealg/__version__.py

@@ -1 +1 @@
-__version__ = "0.1.1"
+__version__ = "0.1.2"

freealg/_util.py

@@ -72,14 +72,14 @@ def force_density(psi0, support, approx, grid, alpha=0.0, beta=0.0):
     if beta <= 0.0 and beta > -0.5:
         constraints.append({
             'type': 'eq',
-            'fun': lambda psi: approx(numpy.array([lam_m], psi))[0]
+            'fun': lambda psi: approx(numpy.array([lam_m]), psi)[0]
         })
 
     # Enforce zero at right edge
     if alpha <= 0.0 and alpha > -0.5:
         constraints.append({
             'type': 'eq',
-            'fun': lambda psi: approx(numpy.array([lam_p], psi))[0]
+            'fun': lambda psi: approx(numpy.array([lam_p]), psi)[0]
         })
 
     # Solve a small quadratic programming
@@ -89,4 +89,12 @@ def force_density(psi0, support, approx, grid, alpha=0.0, beta=0.0):
                    method='SLSQP',
                    options={'maxiter': 1000, 'ftol': 1e-9, 'eps': 1e-8})
 
-    return res.x
+    psi = res.x
+
+    # Normalize first mode to unit mass
+    x = numpy.linspace(lam_m, lam_p, 1000)
+    rho = approx(x, psi)
+    mass = numpy.trapz(rho, x)
+    psi[0] = psi[0] / mass
+
+    return psi

freealg/freeform.py

@@ -477,7 +477,7 @@ class FreeForm(object):
 
         # Check density is unit mass
         mass = numpy.trapz(rho, x)
-        if not numpy.isclose(mass, 1.0, atol=1e-3):
+        if not numpy.isclose(mass, 1.0, atol=1e-2):
             # raise RuntimeWarning(f'"rho" is not unit mass. mass: {mass}. ' +
             #                      r'Set "force=True".')
             print(f'"rho" is not unit mass. mass: {mass}. Set "force=True".')

{freealg-0.1.1.dist-info → freealg-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: freealg
-Version: 0.1.1
+Version: 0.1.2
 Summary: Free probability for large matrices
 Keywords: leaderboard bot chat
 Platform: Linux

{freealg-0.1.1.dist-info → freealg-0.1.2.dist-info}/RECORD

@@ -1,5 +1,5 @@
 freealg/__init__.py,sha256=K92neXJZ9VE1U_j_pj28Qyq1MzlMXhOuYK2ZihgwCaU,463
-freealg/__version__.py,sha256=rnObPjuBcEStqSO0S6gsdS_ot8ITOQjVj_-P1LUUYpg,22
+freealg/__version__.py,sha256=YvuYzWnKtqBb-IqG8HAu-nhIYAsgj9Vmc_b9o7vO-js,22
 freealg/_chebyshev.py,sha256=X6u5pKjR1HPZ-KbCfr7zT6HRwB6pZMADvVS3sT5LTkA,5638
 freealg/_damp.py,sha256=k2vtBtWOxQBf4qXaWu_En81lQBXbEO4QbxxWpvuVhdE,1802
 freealg/_decompress.py,sha256=H7ocq09gQnCY-q_8xHi6qyYj3qp239MCgj406hn0yeE,3344
@@ -7,15 +7,15 @@ freealg/_jacobi.py,sha256=AT4ONSHGGDxVKE3MGMLyMR8uDFiO-e9u3x5udYfdJJk,5635
 freealg/_pade.py,sha256=mP96wEPfIzHLZ6PDB5OyhmSA8N1uVPVUkmJa3ebXXiU,13623
 freealg/_plot_util.py,sha256=aL0u7FHdCwLTj4dZ4SSGA00wab0Voem2nBAsBVvo6XY,18400
 freealg/_sample.py,sha256=K1ZxKoiuPbEKyh-swL5X7gz1kYcQno6Mof0o1xF38tg,2323
-freealg/_util.py,sha256=wJ-t8LMZZFEr2PsZEVqTJP_jQTQ3rHUf0dO27F6L7YQ,2310
-freealg/freeform.py,sha256=MintRRVsEwIrur7KC4tvYa8nBVpNeaajphIqUCmGnLs,27910
+freealg/_util.py,sha256=thQpJcfBNniCH4UzTeQ1yRfXPOVa-y9JkGSdE-wjp00,2492
+freealg/freeform.py,sha256=zjEVTqYT2QzxwE5sElpnlr9mdEK03B_4whysAMWN3X4,27910
 freealg/distributions/__init__.py,sha256=Hnk9bJi4Wy8I_1uuskRyrT2DUpPN1YmBY5uK7XI3U_o,644
 freealg/distributions/kesten_mckay.py,sha256=SFYg2_6_MEX9NGfOIW_rRfM9kgF-bdjQiVS0ENJBemQ,15379
 freealg/distributions/marchenko_pastur.py,sha256=0XHhjw1ZDkigGfjU9lmPRgmwecM1-n6VBkGjRJbxpeY,15869
 freealg/distributions/wachter.py,sha256=cG-3XTP5CHSK5o6SQfa3lqQR__ZibeWZ9AoeMxr3Nnk,15602
 freealg/distributions/wigner.py,sha256=gj3XWK4X7wLsB-inoVpn4pHUUM7NOQdgMGw01LEqkcw,14914
-freealg-0.1.1.dist-info/licenses/LICENSE.txt,sha256=J-EEYEtxb3VVf_Bn1TYfWnpY5lMFIM15iLDDcnaDTPA,1443
-freealg-0.1.1.dist-info/METADATA,sha256=WMUZ2N2ZDfTLNuByAjNyP3e7QpCOF-tyV3ImcwUAZY4,2939
-freealg-0.1.1.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-freealg-0.1.1.dist-info/top_level.txt,sha256=eR2wrgYwDdnnJ9Zf5PruPqe4kQav0GMvRsqct6y00Q8,8
-freealg-0.1.1.dist-info/RECORD,,
+freealg-0.1.2.dist-info/licenses/LICENSE.txt,sha256=J-EEYEtxb3VVf_Bn1TYfWnpY5lMFIM15iLDDcnaDTPA,1443
+freealg-0.1.2.dist-info/METADATA,sha256=pGmBSVA0sY30xyCZ_gPni177js48SXeFKlm0VNEbu_o,2939
+freealg-0.1.2.dist-info/WHEEL,sha256=A8Eltl-h0W-qZDVezsLjjslosEH_pdYC2lQ0JcbgCzs,91
+freealg-0.1.2.dist-info/top_level.txt,sha256=eR2wrgYwDdnnJ9Zf5PruPqe4kQav0GMvRsqct6y00Q8,8
+freealg-0.1.2.dist-info/RECORD,,

{freealg-0.1.1.dist-info → freealg-0.1.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.4.0)
+Generator: setuptools (80.7.0)
 Root-Is-Purelib: true
 Tag: py3-none-any