a-ragab-h-m commited on
Commit
d857e1a
·
verified ·
1 Parent(s): 7d58825

Update Actor/actor.py

Browse files
Files changed (1) hide show
  1. Actor/actor.py +1 -1
Actor/actor.py CHANGED
@@ -509,7 +509,7 @@ class Actor(nn.Module):
509
 
510
  if use_time:
511
  #check for arrival times
512
- too_far = 1 - self.check_arival_times()
513
 
514
  #is depot
515
  a = self.fleet.depot.reshape(self.batch_size, self.num_cars, 1).repeat(1, 1, self.num_nodes)
 
509
 
510
  if use_time:
511
  #check for arrival times
512
+ too_far = torch.logical_not(self.check_arrival_times())
513
 
514
  #is depot
515
  a = self.fleet.depot.reshape(self.batch_size, self.num_cars, 1).repeat(1, 1, self.num_nodes)