Dev Update (#582)
* Fix adaptive refinement (#571) --------- Co-authored-by: Dario Coscia <93731561+dario-coscia@users.noreply.github.com> * Remove collector * Fixes * Fixes * rm unnecessary comment * fix advection (#581) * Fix tutorial .html link (#580) * fix problem data collection for v0.1 (#584) * Message Passing Module (#516) * add deep tensor network block * add interaction network block * add radial field network block * add schnet block * add equivariant network block * fix + tests + doc files * fix egnn + equivariance/invariance tests Co-authored-by: Dario Coscia <dariocos99@gmail.com> --------- Co-authored-by: giovanni <giovanni.canali98@yahoo.it> Co-authored-by: AleDinve <giuseppealessio.d@student.unisi.it> * add type checker (#527) --------- Co-authored-by: Filippo Olivo <filippo@filippoolivo.com> Co-authored-by: Giovanni Canali <115086358+GiovanniCanali@users.noreply.github.com> Co-authored-by: giovanni <giovanni.canali98@yahoo.it> Co-authored-by: AleDinve <giuseppealessio.d@student.unisi.it>
This commit is contained in:
@@ -239,6 +239,22 @@ class PinaTensorDataset(PinaDataset):
|
||||
"""
|
||||
return {k: v["input"] for k, v in self.conditions_dict.items()}
|
||||
|
||||
def update_data(self, new_conditions_dict):
|
||||
"""
|
||||
Update the dataset with new data.
|
||||
This method is used to update the dataset with new data. It replaces
|
||||
the current data with the new data provided in the new_conditions_dict
|
||||
parameter.
|
||||
|
||||
:param dict new_conditions_dict: Dictionary containing the new data.
|
||||
:return: None
|
||||
"""
|
||||
for condition, data in new_conditions_dict.items():
|
||||
if condition in self.conditions_dict:
|
||||
self.conditions_dict[condition].update(data)
|
||||
else:
|
||||
self.conditions_dict[condition] = data
|
||||
|
||||
|
||||
class PinaGraphDataset(PinaDataset):
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user