import torch



# Methods for calculating properties of simulation system
class Observables():
    def __init__(self, observable_dict):
        supported_observables = ['kinetic_energy', 'angles']


    def kinetic_energy(self, ):
import torch



# Methods for calculating properties of simulation system
class Observables():
    def __init__(self, observable_dict):
        supported_observables = ['kinetic_energy', 'angles']


    def kinetic_energy(self, ):
# Methods for calculating properties of simulation system