fix loss and dirichlect BC point generation

Author: levimcclenny

examples/burgers.py

@@ -17,13 +17,13 @@
 Domain.generate_collocation_points(N_f)
 
 
-def func_ic(x, y):
+def func_ic(x):
     return -np.sin(x * math.pi)
 
-init = IC(Domain, [func_ic], var=[['x']])
 
+init = IC(Domain, [func_ic], var=[['x']])
 
-upper_x = dirichlectBC(Domain, val=0.0, var='x',target="upper")
+upper_x = dirichlectBC(Domain, val=0.0, var='x', target="upper")
 
 lower_x = dirichlectBC(Domain, val=0.0, var='x', target="lower")
 

tensordiffeq/__init__.py

@@ -4,7 +4,7 @@
 from . import networks
 from . import plotting
 from . import utils
-from .models import CollocationSolver1D, DiscoveryModel
+from .models import CollocationSolverND, DiscoveryModel
 from .utils import constant, LatinHypercubeSample, tensor
 from .plotting import newfig, get_griddata
 from .helpers import find_L2_error

tensordiffeq/boundaries.py

@@ -58,6 +58,7 @@ def __init__(self, domain, val, var, target):
         super().__init__()
         self.dicts_ = [item for item in self.domain.domaindict if item['identifier'] != self.var]
         self.dict_ = next(item for item in self.domain.domaindict if item["identifier"] == self.var)
+        self.target = self.dict_[var+target]
         self.compile()
 
     def create_input(self):