-
Notifications
You must be signed in to change notification settings - Fork 202
Open
Description
Hi, there is an bug in the SAM code, line 227: https://github.com/FenTechSolutions/CausalDiscoveryToolbox/blob/master/cdt/causality/graph/SAM.py#L227
skeleton = th.from_numpy(skeleton.astype('float32'))
AttributeError: 'Tensor' object has no attribute 'astype'
However, skeleton is already casted to a torch.Tensor in line 510, before it goes to line 227: https://github.com/FenTechSolutions/CausalDiscoveryToolbox/blob/master/cdt/causality/graph/SAM.py#L510
if graph is not None:
skeleton = th.Tensor(nx.adjacency_matrix(graph,
nodelist=list(data.columns)).todense())
Metadata
Metadata
Assignees
Labels
No labels