chriselion

"""
One-off script used to convert demo protos.
This takes the old stacked observation data and moves it to an Observation proto
"""

import pathlib
import logging
import os
from typing import List, Tuple
from mlagents.trainers.buffer import Buffer