code: def restart_agent(
      self, wait_for_amnesia: bool = True, wait_for_agent: bool = True
  ) -> None:
      pass
  
file_path: determined/e2e_tests/tests/cluster/abstract_cluster.py
imports: import abc
  from pathlib import Path
label_desc: Use image node_id 4 for calling the Cluster obj's underlying member method code with example usage: obj.restart_agent(wait_for_amnesia, wait_for_agent) without return types