Merge pull request #423 from stellaHSR/minecraft

Minecraft: add path check
This commit is contained in:
Sirui Hong 2023-10-12 17:59:01 +08:00 committed by GitHub
commit b81ee090e4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -2,6 +2,7 @@
# @Date : 2023/9/23 14:14
# @Author : stellahong (stellahong@fuzhi.ai)
# @Desc :
import os
from typing import Iterable, Dict, Any
from pydantic import BaseModel, Field
import requests
@ -497,6 +498,10 @@ class MinecraftPlayer(SoftwareCompany):
"wait_ticks": 20,
}
)
## add resume round_id
if os.path.exists(f"{CKPT_DIR}/curriculum/round_id.json"):
with open(f"{CKPT_DIR}/curriculum/round_id.json", "r") as f:
round_id = json.load(f)["last_round_id"]
else:
# clear the inventory
self.game_memory.mf_instance.reset(
@ -520,7 +525,10 @@ class MinecraftPlayer(SoftwareCompany):
round_id += 1
# add new task into env and continue
# fixme: update self.task
self.start(task=self.task, round=round_id)
with open(f"{CKPT_DIR}/curriculum/round_id.json", "w") as f:
json.dump({"last_round_id": round_id}, f)
self.start(task=self.task, round=round_id)
logger.info(f"{n_round=}")
self._check_balance()