portal 3.7.0__tar.gz → 3.7.1__tar.gz

{portal-3.7.0/portal.egg-info → portal-3.7.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: portal
-Version: 3.7.0
+Version: 3.7.1
 Summary: Fast and reliable distributed systems in Python
 Home-page: http://github.com/danijar/portal
 Author: Danijar Hafner

{portal-3.7.0 → portal-3.7.1}/portal/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '3.7.0'
+__version__ = '3.7.1'
 
 import multiprocessing as mp
 try:

{portal-3.7.0 → portal-3.7.1}/portal/batching.py

@@ -19,7 +19,8 @@ class BatchServer:
       self, port, name='Server', workers=1, errors=True,
       process=True, shmem=False, **kwargs):
     inner_port = utils.free_port()
-    assert port != inner_port, (port, inner_port)
+    while inner_port == port:
+      inner_port = utils.free_port()
     self.name = name
     self.server = server.Server(inner_port, name, workers, errors, **kwargs)
     if process:

{portal-3.7.0 → portal-3.7.1/portal.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: portal
-Version: 3.7.0
+Version: 3.7.1
 Summary: Fast and reliable distributed systems in Python
 Home-page: http://github.com/danijar/portal
 Author: Danijar Hafner

{portal-3.7.0 → portal-3.7.1}/tests/test_client.py

@@ -286,6 +286,7 @@ class TestClient:
     port = portal.free_port()
     a = threading.Barrier(2)
     b = threading.Barrier(2)
+    c = threading.Barrier(2)
 
     def server():
       server = Server(port)
@@ -296,10 +297,11 @@ class TestClient:
       stats = server.stats()
       assert stats['numrecv'] == 1
       assert stats['numsend'] == stats['numrecv']
+      b.wait()
       server = Server(port)
       server.bind('fn', lambda x: x)
       server.start(block=False)
-      b.wait()
+      c.wait()
       server.close()
 
     def client():
@@ -310,9 +312,10 @@ class TestClient:
       time.sleep(0.1)
       with pytest.raises(portal.Disconnected):
         client.fn(3).result()
+      b.wait()
       client.connect()
       assert client.fn(3).result() == 3
-      b.wait()
+      c.wait()
       client.close()
 
     portal.run([