deepmodeling / dpdispatcher

generate HPC scheduler systems jobs input scripts and submit these scripts to HPC systems and poke until they finish
https://docs.deepmodeling.com/projects/dpdispatcher/
GNU Lesser General Public License v3.0
42 stars 56 forks source link

improve _get_files in SSH context #473

Closed thangckt closed 1 month ago

thangckt commented 1 month ago

this PR solves problem #471 and have some improvements

Summary by CodeRabbit

Summary by CodeRabbit

coderabbitai[bot] commented 1 month ago
Walkthrough ## Walkthrough This update enhances code readability and maintainability across multiple files in the project. Key changes include improved error handling, streamlined command executions, and consistent variable naming. These modifications aim to enhance clarity for future developers without altering the original functionality, thereby simplifying the code structure and improving overall robustness. ## Changes | File Path | Change Summary | |-------------------------------------|------------------------------------------------------------------------------------------------| | `dpdispatcher/contexts/ssh_context.py` | Refactored `write_file` method for better error handling; improved `_get_files` method with streamlined command construction and enhanced error messaging. | | `dpdispatcher/machines/pbs.py` | Renamed `sge_pe_name` to `pe_name` for clarity; updated string formatting to f-strings; revised `resources_subfields` method to include new `job_name` argument and updated documentation accordingly. | ## Sequence Diagram(s) ```mermaid sequenceDiagram participant User participant PBSDispatcher participant SSHContext User->>PBSDispatcher: Request script generation PBSDispatcher->>PBSDispatcher: Generate script header PBSDispatcher->>User: Return script header User->>SSHContext: Check SSH status SSHContext->>SSHContext: Ensure SSH is alive SSHContext->>User: Return status ```

Recent review details **Configuration used: CodeRabbit UI** **Review profile: CHILL**
Commits Files that changed from the base of the PR and between 7bbc4857844d854a6ad7a86fef23a0b3679b61fb and bca9728122a88f684f31cc7b9d155c2a29fda0d4.
Files selected for processing (2) * dpdispatcher/contexts/ssh_context.py (6 hunks) * dpdispatcher/machines/pbs.py (5 hunks)
Additional context used
GitHub Check: codecov/patch
dpdispatcher/machines/pbs.py
[warning] 218-218: dpdispatcher/machines/pbs.py#L218 Added line #L218 was not covered by tests
dpdispatcher/contexts/ssh_context.py
[warning] 298-298: dpdispatcher/contexts/ssh_context.py#L298 Added line #L298 was not covered by tests --- [warning] 812-815: dpdispatcher/contexts/ssh_context.py#L812-L815 Added lines #L812 - L815 were not covered by tests --- [warning] 817-817: dpdispatcher/contexts/ssh_context.py#L817 Added line #L817 was not covered by tests --- [warning] 819-821: dpdispatcher/contexts/ssh_context.py#L819-L821 Added lines #L819 - L821 were not covered by tests --- [warning] 969-970: dpdispatcher/contexts/ssh_context.py#L969-L970 Added lines #L969 - L970 were not covered by tests --- [warning] 972-972: dpdispatcher/contexts/ssh_context.py#L972 Added line #L972 was not covered by tests --- [warning] 976-976: dpdispatcher/contexts/ssh_context.py#L976 Added line #L976 was not covered by tests --- [warning] 981-985: dpdispatcher/contexts/ssh_context.py#L981-L985 Added lines #L981 - L985 were not covered by tests --- [warning] 988-988: dpdispatcher/contexts/ssh_context.py#L988 Added line #L988 was not covered by tests
Additional comments not posted (5)
dpdispatcher/machines/pbs.py (1)
`270-270`: **LGTM!** The string formatting has been updated to f-strings, improving readability.
dpdispatcher/contexts/ssh_context.py (4)
`298-302`: **LGTM!** The additional exception handling improves robustness.
Tools
GitHub Check: codecov/patch
[warning] 298-298: dpdispatcher/contexts/ssh_context.py#L298 Added line #L298 was not covered by tests
--- `812-821`: **LGTM! But verify the function usage in the codebase.** The code changes are approved. However, ensure that all function calls to `write_file` are tested.
Verification successful --- **LGTM! But verify the function usage in the codebase.** The code changes are approved. The function `write_file` is adequately tested and integrated into the codebase, as evidenced by its presence in multiple test files and other parts of the codebase.
Scripts executed The following scripts were executed for the analysis: Script: ```shell #!/bin/bash # Description: Verify all function calls to `write_file` are tested. # Test: Search for the function usage. Expect: Only occurrences of the new signature. rg --type python -A 5 $'write_file' ``` Length of output: 65 --- Script: ```shell #!/bin/bash # Description: Verify all function calls to `write_file` are tested. # Test: Search for the function usage. Expect: Only occurrences of the new signature. rg --type py -A 5 $'write_file' ``` Length of output: 18112
Tools
GitHub Check: codecov/patch
[warning] 812-815: dpdispatcher/contexts/ssh_context.py#L812-L815 Added lines #L812 - L815 were not covered by tests --- [warning] 817-817: dpdispatcher/contexts/ssh_context.py#L817 Added line #L817 was not covered by tests --- [warning] 819-821: dpdispatcher/contexts/ssh_context.py#L819-L821 Added lines #L819 - L821 were not covered by tests
--- `362-373`: **LGTM! But verify the function usage in the codebase.** The code changes are approved. However, ensure that all function calls to `arginfo` are tested. --- `969-989`: **LGTM! But verify the function usage in the codebase.** The code changes are approved. However, ensure that all function calls to `_get_files` are tested.
Verification successful --- **Function usage verified:** The `_get_files` function in `SSHContext` is indirectly tested through the test cases for the `SSHContext` class found in `tests/test_ssh_context.py`. This ensures that the function is covered by tests. - `tests/test_ssh_context.py`
Scripts executed The following scripts were executed for the analysis: Script: ```shell #!/bin/bash # Description: Verify all function calls to `_get_files` are tested. # Test: Search for the function usage. Expect: Only occurrences of the new signature. rg --type python -A 5 $'_get_files' ``` Length of output: 65 --- Script: ```shell #!/bin/bash # Description: Verify all function calls to `_get_files` are tested. # Test: Search for the function usage. Expect: Only occurrences of the new signature. rg --type py -A 5 '_get_files' ``` Length of output: 910 --- Script: ```shell #!/bin/bash # Description: Verify that the `_get_files` function and its caller are covered by tests. # Test: Search for test cases covering `_get_files` or its caller. rg --type py '_get_files' tests/ rg --type py 'self._get_files' tests/ ``` Length of output: 70 --- Script: ```shell #!/bin/bash # Description: Verify that the `SSHContext` class or the module `ssh_context.py` is covered by tests. # Test: Search for test cases covering the `SSHContext` class or the module `ssh_context.py`. rg --type py 'SSHContext' tests/ rg --type py 'ssh_context' tests/ ``` Length of output: 2033
Tools
GitHub Check: codecov/patch
[warning] 969-970: dpdispatcher/contexts/ssh_context.py#L969-L970 Added lines #L969 - L970 were not covered by tests --- [warning] 972-972: dpdispatcher/contexts/ssh_context.py#L972 Added line #L972 was not covered by tests --- [warning] 976-976: dpdispatcher/contexts/ssh_context.py#L976 Added line #L976 was not covered by tests --- [warning] 981-985: dpdispatcher/contexts/ssh_context.py#L981-L985 Added lines #L981 - L985 were not covered by tests --- [warning] 988-988: dpdispatcher/contexts/ssh_context.py#L988 Added line #L988 was not covered by tests
--- Thank you for using CodeRabbit. We offer it for free to the OSS community and would appreciate your support in helping us grow. If you find it useful, would you consider giving us a shout-out on your favorite social media?
Share - [X](https://twitter.com/intent/tweet?text=I%20just%20used%20%40coderabbitai%20for%20my%20code%20review%2C%20and%20it%27s%20fantastic%21%20It%27s%20free%20for%20OSS%20and%20offers%20a%20free%20trial%20for%20the%20proprietary%20code.%20Check%20it%20out%3A&url=https%3A//coderabbit.ai) - [Mastodon](https://mastodon.social/share?text=I%20just%20used%20%40coderabbitai%20for%20my%20code%20review%2C%20and%20it%27s%20fantastic%21%20It%27s%20free%20for%20OSS%20and%20offers%20a%20free%20trial%20for%20the%20proprietary%20code.%20Check%20it%20out%3A%20https%3A%2F%2Fcoderabbit.ai) - [Reddit](https://www.reddit.com/submit?title=Great%20tool%20for%20code%20review%20-%20CodeRabbit&text=I%20just%20used%20CodeRabbit%20for%20my%20code%20review%2C%20and%20it%27s%20fantastic%21%20It%27s%20free%20for%20OSS%20and%20offers%20a%20free%20trial%20for%20proprietary%20code.%20Check%20it%20out%3A%20https%3A//coderabbit.ai) - [LinkedIn](https://www.linkedin.com/sharing/share-offsite/?url=https%3A%2F%2Fcoderabbit.ai&mini=true&title=Great%20tool%20for%20code%20review%20-%20CodeRabbit&summary=I%20just%20used%20CodeRabbit%20for%20my%20code%20review%2C%20and%20it%27s%20fantastic%21%20It%27s%20free%20for%20OSS%20and%20offers%20a%20free%20trial%20for%20proprietary%20code)
Tips ### Chat There are 3 ways to chat with [CodeRabbit](https://coderabbit.ai): - Review comments: Directly reply to a review comment made by CodeRabbit. Example: - `I pushed a fix in commit .` - `Generate unit testing code for this file.` - `Open a follow-up GitHub issue for this discussion.` - Files and specific lines of code (under the "Files changed" tab): Tag `@coderabbitai` in a new review comment at the desired location with your query. Examples: - `@coderabbitai generate unit testing code for this file.` - `@coderabbitai modularize this function.` - PR comments: Tag `@coderabbitai` in a new PR comment to ask questions about the PR branch. For the best results, please provide a very specific query, as very limited context is provided in this mode. Examples: - `@coderabbitai generate interesting stats about this repository and render them as a table.` - `@coderabbitai show all the console.log statements in this repository.` - `@coderabbitai read src/utils.ts and generate unit testing code.` - `@coderabbitai read the files in the src/scheduler package and generate a class diagram using mermaid and a README in the markdown format.` - `@coderabbitai help me debug CodeRabbit configuration file.` Note: Be mindful of the bot's finite context window. It's strongly recommended to break down tasks such as reading entire modules into smaller chunks. For a focused discussion, use review comments to chat about specific files and their changes, instead of using the PR comments. ### CodeRabbit Commands (invoked as PR comments) - `@coderabbitai pause` to pause the reviews on a PR. - `@coderabbitai resume` to resume the paused reviews. - `@coderabbitai review` to trigger an incremental review. This is useful when automatic reviews are disabled for the repository. - `@coderabbitai full review` to do a full review from scratch and review all the files again. - `@coderabbitai summary` to regenerate the summary of the PR. - `@coderabbitai resolve` resolve all the CodeRabbit review comments. - `@coderabbitai configuration` to show the current CodeRabbit configuration for the repository. - `@coderabbitai help` to get help. Additionally, you can add `@coderabbitai ignore` anywhere in the PR description to prevent this PR from being reviewed. ### CodeRabbit Configuration File (`.coderabbit.yaml`) - You can programmatically configure CodeRabbit by adding a `.coderabbit.yaml` file to the root of your repository. - Please see the [configuration documentation](https://docs.coderabbit.ai/guides/configure-coderabbit) for more information. - If your editor has YAML language server enabled, you can add the path at the top of this file to enable auto-completion and validation: `# yaml-language-server: $schema=https://coderabbit.ai/integrations/schema.v2.json` ### Documentation and Community - Visit our [Documentation](https://coderabbit.ai/docs) for detailed information on how to use CodeRabbit. - Join our [Discord Community](https://discord.com/invite/GsXnASn26c) to get help, request features, and share feedback. - Follow us on [X/Twitter](https://twitter.com/coderabbitai) for updates and announcements.
thangckt commented 1 month ago

hi @njzjz I did not see your PR when I try to make some change BTW, this PR can fix the problem

codecov[bot] commented 1 month ago

Codecov Report

Attention: Patch coverage is 4.76190% with 20 lines in your changes missing coverage. Please review.

Project coverage is 47.46%. Comparing base (6251619) to head (bca9728).

Files Patch % Lines
dpdispatcher/contexts/ssh_context.py 0.00% 19 Missing :warning:
dpdispatcher/machines/pbs.py 50.00% 1 Missing :warning:

:exclamation: There is a different number of reports uploaded between BASE (6251619) and HEAD (bca9728). Click for more details.

HEAD has 1 upload less than BASE | Flag | BASE (6251619) | HEAD (bca9728) | |------|------|------| ||15|14|
Additional details and impacted files ```diff @@ Coverage Diff @@ ## master #473 +/- ## =========================================== - Coverage 59.97% 47.46% -12.52% =========================================== Files 39 39 Lines 3843 3845 +2 =========================================== - Hits 2305 1825 -480 - Misses 1538 2020 +482 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.