Skip to content

Commit

Permalink
Sorry about the build fail; I know it is probably for a reason, so I …
Browse files Browse the repository at this point in the history
…will come back to it soon.
  • Loading branch information
Victoria authored and Victoria committed Sep 4, 2024
1 parent dced6d9 commit 420ceea
Show file tree
Hide file tree
Showing 3 changed files with 1 addition and 5 deletions.
3 changes: 1 addition & 2 deletions dgl_ptm/dgl_ptm/model/data_collection.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@ def data_collection(agent_graph,
ndata=None,
edata=None,
format = 'xarray',
initial_only = None,
mode = 'w-'):
"""data_collection - collects data from agents and edges.
Expand Down Expand Up @@ -42,7 +41,7 @@ def data_collection(agent_graph,
ndata = list(agent_graph.node_attr_schemes().keys() - ndata[1])
elif sum(1 for item in ndata if isinstance(item, list)) > 1:
ndata_list = ndata
#initial_only=[]
initial_only=[]
for specification in ndata_list:
if specification == ['all']:
raise ValueError('Use of "all" is not compatible with multiple data collection specification lists.')
Expand Down
1 change: 0 additions & 1 deletion dgl_ptm/dgl_ptm/model/initialize_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -322,7 +322,6 @@ def initialize_model_properties(self):
"""
model_theta = self._initialize_model_theta()
self.steering_parameters['modelTheta'] = model_theta
self.initial_only = []
#attachProb = self._initialize_attach_prob()
#self.steering_parameters['attachProb'] = attachProb

Expand Down
2 changes: 0 additions & 2 deletions dgl_ptm/dgl_ptm/model/step.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,6 @@ def ptm_step(agent_graph, device, timestep, params):
epath = params['epath'],
ndata = params['ndata'],
edata = params['edata'],
initial_only = params['initial_only'],
mode = params['mode']
)
return
Expand Down Expand Up @@ -170,6 +169,5 @@ def ptm_step(agent_graph, device, timestep, params):
epath = params['epath'],
ndata = params['ndata'],
edata = params['edata'],
initial_only = params['initial_only'],
mode = params['mode']
)

0 comments on commit 420ceea

Please sign in to comment.