diff --git a/elastica/rigidbody/cylinder.py b/elastica/rigidbody/cylinder.py index f9078d493..58bdaa9e8 100644 --- a/elastica/rigidbody/cylinder.py +++ b/elastica/rigidbody/cylinder.py @@ -64,8 +64,8 @@ def __init__(self, start, direction, normal, base_length, base_radius, density): self.velocity_collection = np.zeros((MaxDimension.value(), 1)) self.omega_collection = np.zeros((MaxDimension.value(), 1)) - self.acceleration_collection = 0.0 * self.velocity_collection - self.alpha_collection = 0.0 * self.omega_collection + self.acceleration_collection = np.zeros((MaxDimension.value(), 1)) + self.alpha_collection = np.zeros((MaxDimension.value(), 1)) self.director_collection = np.zeros( (MaxDimension.value(), MaxDimension.value(), 1) diff --git a/elastica/rigidbody/sphere.py b/elastica/rigidbody/sphere.py index a02d42591..45dbc5243 100644 --- a/elastica/rigidbody/sphere.py +++ b/elastica/rigidbody/sphere.py @@ -54,8 +54,8 @@ def __init__(self, center, base_radius, density): self.velocity_collection = np.zeros((MaxDimension.value(), 1)) self.omega_collection = np.zeros((MaxDimension.value(), 1)) - self.acceleration_collection = 0.0 * self.velocity_collection - self.alpha_collection = 0.0 * self.omega_collection + self.acceleration_collection = np.zeros((MaxDimension.value(), 1)) + self.alpha_collection = np.zeros((MaxDimension.value(), 1)) self.director_collection = np.zeros( (MaxDimension.value(), MaxDimension.value(), 1)