passing tests locally, remove not implemented checks

This commit is contained in:
evonneng committed 2022-04-09 06:00:33 +00:00
1 parent c8d84937e0
commit cffb48e3e0
2 files changed
+2 -4

No files matched your search

+2 -2
View File
@@ -15,7 +15,7 @@ class GraphInputs:
class Graph(nn.Module):
"""_summary_ """
"""_summary_"""
def __init__(self, modules_config: dict) -> None:
super().__init__()
@@ -39,7 +39,7 @@ class Graph(nn.Module):
Returns:
list: ordering of the module names that should be executed
"""
raise NotImplementedError
return modules_config.keys()
def forward(self, x):
"""_summary_
-2
View File
@@ -12,8 +12,6 @@ class Encoding(nn.Module):
super().__init__()
self.out_dim = out_dim
raise NotImplementedError
def encode(self, in_tensor: TensorType[..., "input_dim"]) -> TensorType[..., "output_dim"]:
"""Encodes an input tensor.