Skip to content

Commit

Permalink
Make IDEs aware of shared regex compilations
Browse files Browse the repository at this point in the history
PyCharm/ruff only looks at changes to self when inspecting code.

* Copy compiled shared attributes into self.
  • Loading branch information
Joppe committed Mar 18, 2024
1 parent 6f8ff3a commit 68dc1fd
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions makemake.py
Original file line number Diff line number Diff line change
Expand Up @@ -466,6 +466,7 @@ def __call__(self, parser, args, values, option_string=None):
if not hasattr(self, 'try_again_in_seconds'):
self.__class__.try_again_in_seconds, self.__class__.try_again_in_milliseconds = map(
re.compile, (self.try_again_in_seconds_pattern, self.try_again_in_milliseconds_pattern))
self.try_again_in_seconds, self.try_again_in_milliseconds = self.__class__.try_again_in_seconds, self.__class__.try_again_in_milliseconds

seconds = self.try_again_in_seconds.findall(r.text)
wait = float(seconds[0]) if seconds else int(self.try_again_in_milliseconds.findall(r.text)[0]) / 1000.
Expand All @@ -481,6 +482,7 @@ def __call__(self, parser, args, values, option_string=None):
(diff, sign, re.compile(r'((^%s.*\n)+)' % compilable_sign, re.MULTILINE).split)
for diff, sign, compilable_sign in [('dropped', '-', '-'), ('added', '+', '\+')]]
self.__class__.maximum_tokens = re.compile(self.maximum_tokens_pattern)
self.diff_splitters, self.maximum_tokens = self.__class__.diff_splitters, self.__class__.maximum_tokens

encoding = tiktoken.encoding_for_model(model)
maximum_tokens = int(self.maximum_tokens.findall(r.text)[0])
Expand Down

0 comments on commit 68dc1fd

Please sign in to comment.