In [16]:

    

%load_ext autoreload
%autoreload 2


    

    




The autoreload extension is already loaded. To reload it, use:
  %reload_ext autoreload

In [17]:

    

import math
import random
import time

from collections import defaultdict


    

In [19]:

    

from ..tf_rl.simulation import DoublePendulum
from tf_rl import simulate


    

    




---------------------------------------------------------------------------
SystemError                               Traceback (most recent call last)
<ipython-input-19-0ae60178164a> in <module>()
----> 1 from ..tf_rl.simulation import DoublePendulum
      2 from tf_rl import simulate

SystemError: Parent module '' not loaded, cannot perform relative import

In [9]:

    

DOUBLE_PENDULUM_PARAMS = {
    'g_ms2': 9.8, # acceleration due to gravity, in m/s^2
    'l1_m': 1.0, # length of pendulum 1 in m
    'l2_m': 2.0, # length of pendulum 2 in m
    'm1_kg': 1.0, # mass of pendulum 1 in kg
    'm2_kg': 1.0, # mass of pendulum 2 in kg
    'damping': 0.4,
    'max_control_input': 20.0
}


    

In [10]:

    

d = DoublePendulum(DOUBLE_PENDULUM_PARAMS)


    

In [11]:

    

d.perform_action(0.2)


    

In [13]:

    

try:
    simulate(d, fps=30, actions_per_simulation_second=1, speed=1.0, simulation_resultion=0.01)
except KeyboardInterrupt:
    print("Interrupted")


    

    















    




Interrupted

In [ ]: