Add headings

movement_primitives/dmp_potential_field.py

@@ -1,4 +1,7 @@
-"""Visualization of DMP as potential field."""
+"""Potential Field of DMP
+======================
+
+Visualization of DMP as potential field."""
 import numpy as np
 from .dmp import (
     dmp_transformation_system, obstacle_avoidance_acceleration_2d, phase)

movement_primitives/plot.py

@@ -1,4 +1,8 @@
-"""Tools for plotting multi-dimensional trajectories."""
+"""Plotting Tools
+==============
+
+Tools for plotting multi-dimensional trajectories.
+"""
 import matplotlib.pyplot as plt
 
 

movement_primitives/spring_damper.py

@@ -1,4 +1,8 @@
-"""Spring-damper based attractors."""
+"""Spring-Damper Based Attractors
+==============================
+
+Spring-damper based attractors are the basis of a DMP's transformation system.
+"""
 import numpy as np
 import pytransform3d.rotations as pr
 from .base import PointToPointMovement

movement_primitives/visualization.py

@@ -1,4 +1,8 @@
-"""3D visualization tools for movement primitives."""
+"""Visualization Tools
+===================
+
+3D visualization tools for movement primitives.
+"""
 import numpy as np
 import open3d as o3d
 import scipy as sp