Skip to content

Commit 2b14563

Browse files
Update size parameter to accept callable
Allow 'size' parameter to be a callable that returns an int.
1 parent dcb167e commit 2b14563

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

fsspec/callbacks.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,8 +91,10 @@ def set_size(self, size):
9191
9292
Parameters
9393
----------
94-
size: int
94+
size: int or callable
9595
"""
96+
if callable(size):
97+
size = size()
9698
self.size = size
9799
self.call()
98100

0 commit comments

Comments
 (0)