fixed tests

tilmantroester · Oct 29, 2020 · 124b098 · 124b098
1 parent 0f84c43
commit 124b098
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -25,7 +25,7 @@
 author = 'Minas Karamanis'
 
 # The full version, including alpha/beta/rc tags
-release = '2.1.0'
+release = '2.1.1'
 
 # -- General configuration ---------------------------------------------------
 

diff --git a/docs/index.rst b/docs/index.rst
@@ -96,7 +96,7 @@ Copyright 2019-2020 Minas Karamanis and contributors.
 Changelog
 =========
 
-**2.1.0 (29/10/20)**
+**2.1.1 (29/10/20)**
 
 - Added ``blobs`` interface to track arbitrary metadata.
 - Updated ``GlobalMove`` and multimodal example.

diff --git a/setup.py b/setup.py
@@ -8,7 +8,7 @@
 
 setuptools.setup(
     name="zeus-mcmc",
-    version="2.1.0",
+    version="2.1.1",
     author="Minas Karamanis",
     author_email="[email protected]",
     description="zeus: Lightning Fast MCMC",

diff --git a/zeus/tests/test_samples.py b/zeus/tests/test_samples.py
@@ -10,11 +10,11 @@ def test_chain(seed=42):
     ndim = np.random.randint(2,200)
     nwalkers = 2 * ndim
     s = samples(ndim, nwalkers)
-    s.extend(nsteps)
+    s.extend(nsteps, None)
     for i in range(nsteps):
         x = np.random.rand(nwalkers,ndim)
         z = np.random.rand(nwalkers)
-        s.save(x, z)
+        s.save(x, z, None)
     assert np.shape(s.chain) == (nsteps,nwalkers,ndim)
     assert np.shape(s.logprob) == (nsteps,nwalkers)
 
@@ -25,11 +25,11 @@ def test_flatten(seed=42):
     ndim = np.random.randint(2,200)
     nwalkers = 2 * ndim
     s = samples(ndim,nwalkers)
-    s.extend(nsteps)
+    s.extend(nsteps, None)
     for i in range(nsteps):
         x = np.random.rand(nwalkers,ndim)
         z = np.random.rand(nwalkers)
-        s.save(x, z)
+        s.save(x, z, None)
     assert np.shape(s.flatten()) == (nsteps*nwalkers,ndim)
     burn = np.random.randint(2,100)
     thin = np.random.randint(1,10)