Skip to content

Commit bc9828a

Browse files
committed
Fix README
1 parent 5ccab93 commit bc9828a

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

detail/to_PyG.md

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,8 @@ You have to install the `Pytorch` and `PyTorch Geometric` first:
1616
### Constructor
1717

1818
```python
19-
TrussHeteroDataCreator(metapathType: MetapathType, taskType: TaskType) -> None
19+
TrussHeteroDataCreator(metapathType: MetapathType = MetapathType.NO_IMPLICIT,
20+
taskType : TaskType = TaskType.OPTIMIZATION ) -> None
2021
```
2122

2223
- **`metapathType`** : Whether to use impicit connection of the bipartite graph representation of the truss.
@@ -37,7 +38,10 @@ TrussHeteroDataCreator(metapathType: MetapathType, taskType: TaskType) -> None
3738

3839
```python
3940
TrussHeteroDataCreator.FromJSON(
40-
trussJSONFile, trussDim, forceScale, displaceScale, positionScale,
41+
trussJSONFile, trussDim,
42+
forceScale=1.,
43+
displaceScale=1.,
44+
positionScale=1.,
4145
usedMemberTypes=None,
4246
fixedMemberType=MemberType(1., 1e7, 0.1),
4347
isUseFixed=True,
@@ -61,7 +65,10 @@ TrussHeteroDataCreator.FromJSON(
6165

6266
```python
6367
TrussHeteroDataCreator.FromTruss(
64-
truss, forceScale, displaceScale, positionScale,
68+
truss,
69+
forceScale=1.,
70+
displaceScale=1.,
71+
positionScale=1.,
6572
usedMemberTypes=None,
6673
fixedMemberType=MemberType(1., 1e7, 0.1),
6774
isUseFixed=True,

0 commit comments

Comments
 (0)