Update for new RouterAsyncFS import/interface in recent Hail versions

gnomad/utils/file_utils.py

@@ -11,10 +11,15 @@
 from typing import Callable, List, Dict, Optional, Tuple, Union
 
 import hail as hl
-from hailtop.aiotools import LocalAsyncFS, RouterAsyncFS, AsyncFS
+from hailtop.aiotools import LocalAsyncFS, AsyncFS
 from hailtop.aiogoogle import GoogleStorageAsyncFS
 from hailtop.utils import bounded_gather, tqdm
 
+try:
+    from hailtop.aiotools.router_fs import RouterAsyncFS
+except ImportError:
+    from hailtop.aiotools import RouterAsyncFS
+
 logging.basicConfig(format="%(levelname)s (%(name)s %(lineno)s): %(message)s")
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.INFO)
@@ -57,7 +62,7 @@ async def async_file_exists(fs: AsyncFS, fpath: str) -> bool:
     ) as pbar:
         with ThreadPoolExecutor() as thread_pool:
             async with RouterAsyncFS(
-                "file", [LocalAsyncFS(thread_pool), GoogleStorageAsyncFS()]
+                "file", filesystems=[LocalAsyncFS(thread_pool), GoogleStorageAsyncFS()]
             ) as fs:
 
                 def check_existence_and_update_pbar_thunk(fpath: str) -> Callable: