wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix

This commit is contained in:
Remi Cadene 2024-11-27 16:45:07 +01:00
parent a97c1cb1af
commit aa03a27f0a
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lerobot/scripts/visualize_dataset_html.py
View File

@ -88,7 +88,7 @@ def run_server(
port: str, port: str,
static_folder: Path, static_folder: Path,
template_folder: Path, template_folder: Path,
has_policy = False, has_policy=False,
): ):
app = Flask(__name__, static_folder=static_folder.resolve(), template_folder=template_folder.resolve()) app = Flask(__name__, static_folder=static_folder.resolve(), template_folder=template_folder.resolve())
app.config["SEND_FILE_MAX_AGE_DEFAULT"] = 0 # specifying not to cache app.config["SEND_FILE_MAX_AGE_DEFAULT"] = 0 # specifying not to cache
@ -131,7 +131,7 @@ def run_server(
dataset_info=dataset_info, dataset_info=dataset_info,
videos_info=videos_info, videos_info=videos_info,
ep_csv_url=ep_csv_url, ep_csv_url=ep_csv_url,
has_policy = has_policy, has_policy=has_policy,
) )
app.run(host=host, port=port) app.run(host=host, port=port)
@ -404,7 +404,7 @@ def main():
args = parser.parse_args() args = parser.parse_args()
kwargs = vars(args) kwargs = vars(args)
repo_id = kwargs.pop("repo_id") repo_id = kwargs.pop("repo_id")
# root = kwargs.pop("root") kwargs.pop("root")
dataset = LeRobotDataset(repo_id) dataset = LeRobotDataset(repo_id)
visualize_dataset_html(dataset, **kwargs) visualize_dataset_html(dataset, **kwargs)