diff --git a/papermill/iorw.py b/papermill/iorw.py index 14a0122c..9cae0117 100644 --- a/papermill/iorw.py +++ b/papermill/iorw.py @@ -49,6 +49,7 @@ try: from pyarrow.fs import FileSelector, HadoopFileSystem except ImportError: + FileSelector = missing_dependency_generator("pyarrow", "hdfs") HadoopFileSystem = missing_dependency_generator("pyarrow", "hdfs") try: