update pydens

Author: akoryagin

pydens/model_torch.py

@@ -9,7 +9,7 @@
 from torch.autograd import grad
 from tqdm import tqdm
 
-from .batchflow.batchflow.models.torch.layers import ConvBlock # pylint: disable=import-error
+from batchflow.models.torch.layers import Block # pylint: disable=import-error
 
 
 current_model = ContextVar("current_model")
@@ -165,7 +165,7 @@ def __init__(self, ndims, initial_condition=None, boundary_condition=None, domai
 
         # Assemble conv-block.
         fake_inputs = torch.rand((2, self.total), dtype=torch.float32)
-        self.conv_block = ConvBlock(inputs=fake_inputs, **kwargs)
+        self.conv_block = Block(inputs=fake_inputs, **kwargs)
 
     def forward(self, xs):
         u = self.conv_block(xs)
@@ -260,7 +260,7 @@ def pde(f, x, e):
         go into the `model`-class.
 
         Note:
-        `ConvBlockModel` is based on `ConvBlock` from framework
+        `ConvBlockModel` is based on `Block` from framework
         "`BatchFlow <https://github.com/analysiscenter/batchflow>`_".
 
     constraints : sequence or callable

setup.py

@@ -31,6 +31,7 @@
         'tqdm>=4.19.7',
         'scipy>=0.19.1',
         'scikit-image>=0.13.1',
+        'batchflow>=0.8.0',
         'numba>=0.42',
     ],
     classifiers=[