@@ -80,8 +80,8 @@ def get_context(self):
8080class ThreadPoolMixin (ExecutorMixin ):
8181 executor_type = futures .ThreadPoolExecutor
8282
83- def create_barrier (self , count ):
84- return threading .Barrier (count )
83+ def create_barrier (self , count , ** kwargs ):
84+ return threading .Barrier (count , ** kwargs )
8585
8686 def create_event (self ):
8787 return threading .Event ()
@@ -91,7 +91,7 @@ def create_event(self):
9191class InterpreterPoolMixin (ExecutorMixin ):
9292 executor_type = futures .InterpreterPoolExecutor
9393
94- def create_barrier (self , count ):
94+ def create_barrier (self , count , ** kwargs ):
9595 self .skipTest ("InterpreterPoolExecutor doesn't support barriers" )
9696
9797 def create_event (self ):
@@ -113,8 +113,8 @@ def get_context(self):
113113 self .skipTest ("TSAN doesn't support threads after fork" )
114114 return super ().get_context ()
115115
116- def create_barrier (self , count ):
117- return self .manager .Barrier (count )
116+ def create_barrier (self , count , ** kwargs ):
117+ return self .manager .Barrier (count , ** kwargs )
118118
119119 def create_event (self ):
120120 return self .manager .Event ()
@@ -131,8 +131,8 @@ def get_context(self):
131131 self .skipTest ("ProcessPoolExecutor unavailable on this system" )
132132 return super ().get_context ()
133133
134- def create_barrier (self , count ):
135- return self .manager .Barrier (count )
134+ def create_barrier (self , count , ** kwargs ):
135+ return self .manager .Barrier (count , ** kwargs )
136136
137137 def create_event (self ):
138138 return self .manager .Event ()
@@ -153,8 +153,8 @@ def get_context(self):
153153 self .skipTest ("TSAN doesn't support threads after fork" )
154154 return super ().get_context ()
155155
156- def create_barrier (self , count ):
157- return self .manager .Barrier (count )
156+ def create_barrier (self , count , ** kwargs ):
157+ return self .manager .Barrier (count , ** kwargs )
158158
159159 def create_event (self ):
160160 return self .manager .Event ()
0 commit comments