Imports and function decleration

MiniSnakes.py

import torch as t
from torch import tensor as T
from numpy import unravel_index as unravel
import matplotlib.pyplot as plt

def do(snake, action):
   '''This is where the magic happens :) '''