Fixes for install_updates

albertyw · Feb 5, 2024 · 75f0ccc · 75f0ccc
1 parent 9c4852d
commit 75f0ccc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/req_update/python.py b/req_update/python.py
@@ -206,9 +206,9 @@ def install_updates(self) -> None:
             elif updated_file in PYPROJECT_FILES:
                 command = ['pip', 'install', '-e', '.']
                 self.util.execute_shell(command, False)
-                with open('pyproject.toml', 'rb') as handle:
+                with open(updated_file, 'rb') as handle:
                     data = tomllib.load(handle)
-                    optionals = data['project']['optional-dependencies'].keys()
+                    optionals = data['project'].get('optional-dependencies', {}).keys()
                     for optional in optionals:
                         command = ['pip', 'install', '-e', '.[%s]' % optional]
                         self.util.execute_shell(command, False)