fatmacankara commited on
Commit
6259ee3
1 Parent(s): 771222f

Update code/add_domains.py

Browse files
Files changed (1) hide show
  1. code/add_domains.py +1 -1
code/add_domains.py CHANGED
@@ -24,7 +24,7 @@ def add_domains(data, path_to_domains):
24
  NotZeroDistance = data[data.distance != 0.0]
25
  NotZeroDistance.distance = -1000
26
 
27
- NotZeroDistance = NotZeroDistance[~NotZeroDistance.domain.isin(zeroDistanceDomains)]
28
 
29
  data = pd.concat([ZeroDistance, NotZeroDistance], sort=False)
30
  data.reset_index(drop=True, inplace=True)
 
24
  NotZeroDistance = data[data.distance != 0.0]
25
  NotZeroDistance.distance = -1000
26
 
27
+ NotZeroDistance = NotZeroDistance[~NotZeroDistance.datapoint.isin(ZeroDistance.datapoint.to_list())]
28
 
29
  data = pd.concat([ZeroDistance, NotZeroDistance], sort=False)
30
  data.reset_index(drop=True, inplace=True)