diff --git a/docker/Dockerfile.benchmark b/docker/Dockerfile.benchmark index 7c05c9e1f..75df1f810 100644 --- a/docker/Dockerfile.benchmark +++ b/docker/Dockerfile.benchmark @@ -133,7 +133,8 @@ assets_dst = os.path.join(get_libero_path('benchmark_root'), 'assets'); \ print(f'Downloading LIBERO-plus assets to {assets_dst}...'); \ zp = hf_hub_download('Sylvest/LIBERO-plus', 'assets.zip', repo_type='dataset', local_dir='/tmp/lp-dl'); \ zipfile.ZipFile(zp).extractall('/tmp/lp-unzip'); \ -src = glob.glob('/tmp/lp-unzip/inspire/*/assets')[0]; \ +hits = glob.glob('/tmp/lp-unzip/**/assets/scenes', recursive=True); \ +src = os.path.dirname(hits[0]); \ shutil.move(src, assets_dst); \ shutil.rmtree('/tmp/lp-dl', True); shutil.rmtree('/tmp/lp-unzip', True); \ print(f'Assets installed: {os.listdir(assets_dst)[:5]}...')" ;; \ diff --git a/docker/Dockerfile.eval-libero-plus b/docker/Dockerfile.eval-libero-plus index 80e38b8bb..4f312468e 100644 --- a/docker/Dockerfile.eval-libero-plus +++ b/docker/Dockerfile.eval-libero-plus @@ -51,7 +51,8 @@ assets_dst = os.path.join(get_libero_path('benchmark_root'), 'assets'); \ print(f'Downloading LIBERO-plus assets to {assets_dst}...'); \ zp = hf_hub_download('Sylvest/LIBERO-plus', 'assets.zip', repo_type='dataset', local_dir='/tmp/lp-dl'); \ zipfile.ZipFile(zp).extractall('/tmp/lp-unzip'); \ -src = glob.glob('/tmp/lp-unzip/inspire/*/assets')[0]; \ +hits = glob.glob('/tmp/lp-unzip/**/assets/scenes', recursive=True); \ +src = os.path.dirname(hits[0]); \ shutil.move(src, assets_dst); \ shutil.rmtree('/tmp/lp-dl', True); shutil.rmtree('/tmp/lp-unzip', True); \ print(f'Assets installed: {os.listdir(assets_dst)[:5]}...')"