folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve conditions and refactor dataset classes (#475)

Browse Source 
* Reimplement conditions

* Refactor datasets and implement LabelBatch

---------

Co-authored-by: Dario Coscia <dariocos99@gmail.com>
This commit is contained in:
Filippo Olivo
2025-03-07 11:24:09 +01:00
committed by Nicola Demo
parent bdad144461
commit a0cbf1c44a
40 changed files with 943 additions and 550 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
pina/condition/domain_equation_condition.py
View File

@@ -1,4 +1,6 @@
import torch
"""
DomainEquationCondition class definition.
"""
from .condition_interface import ConditionInterface
from ..utils import check_consistency
@@ -16,7 +18,11 @@ class DomainEquationCondition(ConditionInterface):
def __init__(self, domain, equation):
"""
TODO : add docstring
Initialize the DomainEquationCondition, storing the domain and equation.
:param DomainInterface domain: Domain object containing the domain data
:param EquationInterface equation: Equation object containing the
equation data
"""
super().__init__()
self.domain = domain
@@ -29,5 +35,5 @@ class DomainEquationCondition(ConditionInterface):
elif key == "equation":
check_consistency(value, (EquationInterface))
DomainEquationCondition.__dict__[key].__set__(self, value)
elif key in ("_problem", "_condition_type"):
elif key in ("_problem"):
super().__setattr__(key, value)