Skip to content

Commit

Permalink
Added static method decorators
Browse files Browse the repository at this point in the history
  • Loading branch information
Gaurav S Deshmukh committed Sep 19, 2023
1 parent 63b2ad4 commit 16161e3
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/featurizers.py
Original file line number Diff line number Diff line change
Expand Up @@ -568,6 +568,7 @@ class SurfaceBondDistanceFeaturizer(BondDistanceFeaturizer):
def __init__(self, encoder, min=0, max=5, n_intervals=10):
super().__init__(encoder, min=min, max=max, n_intervals=n_intervals)

@staticmethod
def name():
"""Return the name of the featurizer."""
return "surface_bond_distance"
Expand All @@ -584,6 +585,7 @@ class AdsorbateBondDistanceFeaturizer(BondDistanceFeaturizer):
def __init__(self, encoder, min=0, max=4, n_intervals=16):
super().__init__(encoder, min=min, max=max, n_intervals=n_intervals)

@staticmethod
def name():
"""Return the name of the featurizer."""
return "adsorbate_distance"
Expand Down

0 comments on commit 16161e3

Please sign in to comment.