portal 3.2.1__tar.gz → 3.2.2__tar.gz

{portal-3.2.1/portal.egg-info → portal-3.2.2}/PKG-INFO

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

{portal-3.2.1 → portal-3.2.2}/portal/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '3.2.1'
+__version__ = '3.2.2'
 
 import multiprocessing as mp
 try:

{portal-3.2.1 → portal-3.2.2}/portal/contextlib.py

@@ -4,6 +4,7 @@ import pathlib
 import sys
 import threading
 import traceback
+import warnings
 
 import cloudpickle
 import psutil
@@ -143,7 +144,12 @@ class Context:
       return
     if hasattr(worker, 'thread'):
       assert current != worker.thread
-    current.children.append(worker)
+    try:
+      current.children.append(worker)
+    except AttributeError:
+      warnings.warn(
+          'Using Portal from plain Python threads is discouraged because ' +
+          'they can cause hangs during shutdown.')
 
   def children(self, thread):
     current = thread or threading.current_thread()

{portal-3.2.1 → portal-3.2.2}/portal/utils.py

@@ -8,6 +8,8 @@ import time
 
 import psutil
 
+from . import contextlib
+
 
 def run(workers, duration=None):
   [None if x.started else x.start() for x in workers]
@@ -96,9 +98,13 @@ def free_port():
   # Return a port that is currently free. This function is not thread or
   # process safe, because there is no way to guarantee that the port will still
   # be free at the time it will be used.
-  sock = socket.socket()
+  if contextlib.context.serverkw.get('ipv6', False):
+    family, addr = socket.AF_INET6, ('localhost', 0, 0, 0)
+  else:
+    family, addr = socket.AF_INET, ('localhost', 0)
+  sock = socket.socket(family, socket.SOCK_STREAM)
   sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
-  sock.bind(('localhost', 0))
+  sock.bind(addr)
   port = sock.getsockname()[1]
   sock.close()
   return port

{portal-3.2.1 → portal-3.2.2/portal.egg-info}/PKG-INFO

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

{portal-3.2.1 → portal-3.2.2}/tests/test_server.py

@@ -199,7 +199,7 @@ class TestServer:
   def test_shared_pool(self, repeat, Server):
 
     def slow(x):
-      time.sleep(0.2)
+      time.sleep(0.5)
       return x
 
     def fast(x):
@@ -226,7 +226,7 @@ class TestServer:
   def test_separate_pools(self, repeat, Server):
 
     def slow(x):
-      time.sleep(0.1)
+      time.sleep(0.5)
       return x
 
     def fast(x):