Assign turbine cluster power curve to object

Author: SabineHaas

windpowerlib/wind_turbine_cluster.py

@@ -147,8 +147,7 @@ def assign_power_curve(self, wake_losses_model='wind_efficiency_curve',
 
         Returns
         -------
-        cluster_power_curve : pd.DataFrame
-            Calculated power curve of the wind turbine cluster.
+        self
 
         """
         # Assign wind farm power curves to wind farms of wind turbine cluster
@@ -174,4 +173,5 @@ def assign_power_curve(self, wake_losses_model='wind_efficiency_curve',
         cluster_power_curve.columns = ['power']
         # Return wind speed (index) to a column of the data frame
         cluster_power_curve.reset_index('wind_speed', inplace=True)
-        return cluster_power_curve
+        self.power_curve = cluster_power_curve
+        return self