Fix: Resolve PermissionError and UnicodeDecodeError in Python scripts (#1980)
* Fix: Resolve PermissionError and UnicodeDecodeError in Python scripts Problem: 1. PermissionError when running eval.py 2. UnicodeDecodeError: 'gbk' when running migrate_policy_normalization.py * To explicitly specify the file encoding and resolve linter warnings. Signed-off-by: Jivin.L <45867423+JivinDotL@users.noreply.github.com> --------- Signed-off-by: Jivin.L <45867423+JivinDotL@users.noreply.github.com> Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
This commit is contained in:
@@ -196,11 +196,10 @@ class PreTrainedConfig(draccus.ChoiceRegistry, HubMixin, abc.ABC):
|
|||||||
config = json.load(f)
|
config = json.load(f)
|
||||||
|
|
||||||
config.pop("type")
|
config.pop("type")
|
||||||
with tempfile.NamedTemporaryFile("w+") as f:
|
with tempfile.NamedTemporaryFile("w+", delete=False, suffix=".json") as f:
|
||||||
json.dump(config, f)
|
json.dump(config, f)
|
||||||
config_file = f.name
|
config_file = f.name
|
||||||
f.flush()
|
|
||||||
|
|
||||||
cli_overrides = policy_kwargs.pop("cli_overrides", [])
|
cli_overrides = policy_kwargs.pop("cli_overrides", [])
|
||||||
with draccus.config_type("json"):
|
with draccus.config_type("json"):
|
||||||
return draccus.parse(orig_config.__class__, config_file, args=cli_overrides)
|
return draccus.parse(orig_config.__class__, config_file, args=cli_overrides)
|
||||||
|
|||||||
@@ -246,7 +246,9 @@ class PreTrainedPolicy(nn.Module, HubMixin, abc.ABC):
|
|||||||
base_model=base_model,
|
base_model=base_model,
|
||||||
)
|
)
|
||||||
|
|
||||||
template_card = files("lerobot.templates").joinpath("lerobot_modelcard_template.md").read_text()
|
template_card = (
|
||||||
|
files("lerobot.templates").joinpath("lerobot_modelcard_template.md").read_text(encoding="utf-8")
|
||||||
|
)
|
||||||
card = ModelCard.from_template(card_data, template_str=template_card)
|
card = ModelCard.from_template(card_data, template_str=template_card)
|
||||||
card.validate()
|
card.validate()
|
||||||
return card
|
return card
|
||||||
|
|||||||
Reference in New Issue
Block a user