tangger/openpi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

load droid sim eval policies without credentials (#440)

Browse Source 
small change to enable loading from the openpi sim eval bucket without credentials (for joint pos policies)
This commit is contained in:
Karl Pertsch
2025-04-17 15:39:53 -04:00
committed by GitHub
parent 650b02e4ca fe5d5580a4
commit c23bc86a0a
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/openpi/shared/download.py
View File

@@ -93,7 +93,7 @@ def maybe_download(url: str, *, force_download: bool = False, **kwargs) -> pathl
logger.info(f"Downloading {url} to {local_path}") logger.info(f"Downloading {url} to {local_path}")
scratch_path = local_path.with_suffix(".partial") scratch_path = local_path.with_suffix(".partial")
if _is_openpi_url(url): if _is_openpi_url(url) or _is_openpi_simeval_url(url):
# Download without credentials. # Download without credentials.
_download_boto3( _download_boto3(
url, url,
@@ -299,6 +299,9 @@ def _is_openpi_url(url: str) -> bool:
"""Check if the url is an OpenPI S3 bucket url.""" """Check if the url is an OpenPI S3 bucket url."""
return url.startswith("s3://openpi-assets/") return url.startswith("s3://openpi-assets/")
def _is_openpi_simeval_url(url: str) -> bool:
"""Check if the url is an OpenPI simeval S3 bucket url."""
return url.startswith("s3://openpi-assets-simeval/")
def _get_mtime(year: int, month: int, day: int) -> float: def _get_mtime(year: int, month: int, day: int) -> float:
"""Get the mtime of a given date at midnight UTC.""" """Get the mtime of a given date at midnight UTC."""