Skip to content

Commit

Permalink
support spacing in inspector
Browse files Browse the repository at this point in the history
  • Loading branch information
minhuanli committed Aug 2, 2024
1 parent f1afaaf commit 01e7fc6
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions SFC_Torch/Fmodel.py
Original file line number Diff line number Diff line change
Expand Up @@ -538,7 +538,7 @@ def assign_resolution_bins(
if return_labels:
return self.bin_labels

def inspect_data(self, verbose=False):
def inspect_data(self, verbose=False, spacing=4.5):
"""
Do an inspection of data, for hints about
1. solvent percentage for mask calculation
Expand All @@ -547,7 +547,7 @@ def inspect_data(self, verbose=False):
# solvent percentage
vdw_rad = vdw_rad_tensor(self.atom_name, device=self.device)
uc_grid_orth_tensor = unitcell_grid_center(
self.unit_cell, spacing=4.5, return_tensor=True, device=self.device
self.unit_cell, spacing=spacing, return_tensor=True, device=self.device
)
occupancy, _ = packingscore_voxelgrid_torch(
self.atom_pos_orth,
Expand Down

0 comments on commit 01e7fc6

Please sign in to comment.