27 lines
669 B
Python
27 lines
669 B
Python
import random
|
|
|
|
|
|
class MLPlay:
|
|
def __init__(self, *args, **kwargs):
|
|
print("Initial ml script")
|
|
self.count=0
|
|
|
|
def update(self, scene_info: dict, *args, **kwargs):
|
|
"""
|
|
Generate the command according to the received scene information
|
|
"""
|
|
# print("AI received data from game :", json.dumps(scene_info))
|
|
# print(scene_info)
|
|
actions = ["UP", "DOWN", "LEFT", "RIGHT"]
|
|
self.count+=1
|
|
if self.count >100:
|
|
a = actions[100]
|
|
return random.sample(actions, 1)
|
|
|
|
def reset(self):
|
|
"""
|
|
Reset the status
|
|
"""
|
|
print("reset ml script")
|
|
pass
|