Skip to content

Commit

Permalink
Merge branch 'pandas_fix' into develop
Browse files Browse the repository at this point in the history
  • Loading branch information
l-kotzur committed Jun 28, 2023
2 parents 66ce77e + 04a2036 commit fea8ca0
Show file tree
Hide file tree
Showing 6 changed files with 12 additions and 12 deletions.
8 changes: 4 additions & 4 deletions test/test_cluster_order.py
Expand Up @@ -81,19 +81,19 @@ def test_cluster_order():

# sort the typical days in order to avoid error assertion due to different order
sortedDaysOrig1 = (
orig_raw_predefClusterOrder.sum(axis=0, level=0).sort_values("GHI").index
orig_raw_predefClusterOrder.groupby(level=0).sum().sort_values("GHI").index
)
sortedDaysTest1 = (
typPeriods_predefClusterOrder.sum(axis=0, level=0).sort_values("GHI").index
typPeriods_predefClusterOrder.groupby(level=0).sum().sort_values("GHI").index
)

sortedDaysOrig2 = (
orig_raw_predefClusterOrderAndClusterCenters.sum(axis=0, level=0)
orig_raw_predefClusterOrderAndClusterCenters.groupby(level=0).sum()
.sort_values("GHI")
.index
)
sortedDaysTest2 = (
typPeriods_predefClusterOrderAndClusterCenters.sum(axis=0, level=0)
typPeriods_predefClusterOrderAndClusterCenters.groupby(level=0).sum()
.sort_values("GHI")
.index
)
Expand Down
4 changes: 2 additions & 2 deletions test/test_hierarchical.py
Expand Up @@ -42,8 +42,8 @@ def test_hierarchical():
print("Clustering took " + str(time.time() - starttime))

# sort the typical days in order to avoid error assertion due to different order
sortedDaysOrig = orig_raw.sum(axis=0, level=0).sort_values("GHI").index
sortedDaysTest = typPeriods.sum(axis=0, level=0).sort_values("GHI").index
sortedDaysOrig = orig_raw.groupby(level=0).sum().sort_values("GHI").index
sortedDaysTest = typPeriods.groupby(level=0).sum().sort_values("GHI").index

# rearange their order
orig = orig_raw[typPeriods.columns].unstack().loc[sortedDaysOrig, :].stack()
Expand Down
4 changes: 2 additions & 2 deletions test/test_k_medoids.py
Expand Up @@ -39,8 +39,8 @@ def test_k_medoids():
print("Clustering took " + str(time.time() - starttime))

# sort the typical days in order to avoid error assertion due to different order
sortedDaysOrig = orig_raw.sum(axis=0, level=0).sort_values("GHI").index
sortedDaysTest = typPeriods.sum(axis=0, level=0).sort_values("GHI").index
sortedDaysOrig = orig_raw.groupby(level=0).sum().sort_values("GHI").index
sortedDaysTest = typPeriods.groupby(level=0).sum().sort_values("GHI").index

# rearange their order
orig = orig_raw[typPeriods.columns].unstack().loc[sortedDaysOrig, :].stack()
Expand Down
4 changes: 2 additions & 2 deletions test/test_segmentation.py
Expand Up @@ -41,8 +41,8 @@ def test_segmentation():
print("Clustering took " + str(time.time() - starttime))

# sort the typical days in order to avoid error assertion due to different order
sortedDaysOrig = orig_raw.sum(axis=0, level=0).sort_values("GHI").index
sortedDaysTest = typPeriods.sum(axis=0, level=0).sort_values("GHI").index
sortedDaysOrig = orig_raw.groupby(level=0).sum().sort_values("GHI").index
sortedDaysTest = typPeriods.groupby(level=0).sum().sort_values("GHI").index

# rearange their order
orig = orig_raw[typPeriods.columns].unstack().loc[sortedDaysOrig, :].stack()
Expand Down
2 changes: 1 addition & 1 deletion tsam/periodAggregation.py
Expand Up @@ -49,7 +49,7 @@ def aggregatePeriods(
clusterOrder.append(
[n_clusters - 1] * int(n_sets - cluster_size * n_clusters)
)
clusterOrder = np.hstack(np.array(clusterOrder))
clusterOrder = np.hstack(np.array(clusterOrder, dtype=object))
clusterCenters, clusterCenterIndices = representations(
candidates,
clusterOrder,
Expand Down
2 changes: 1 addition & 1 deletion tsam/timeseriesaggregation.py
Expand Up @@ -61,7 +61,7 @@ def unstackToPeriods(timeSeries, timeStepsPerPeriod):
rep_data = unstackedTimeSeries.head(attached_timesteps)

# append them at the end of the time series
unstackedTimeSeries = unstackedTimeSeries.append(rep_data, ignore_index=False)
unstackedTimeSeries = pd.concat([unstackedTimeSeries, rep_data])

# create period and step index
for ii in range(0, len(unstackedTimeSeries)):
Expand Down

0 comments on commit fea8ca0

Please sign in to comment.