BasedLukas

import numpy as np
import random
import matplotlib.pyplot as plt
from typing import Dict, Optional, Tuple


def argmax_a(state: str, q: Dict) -> int:
    """
    Find the action that maximizes the Q-value in a given state.