Implement Local UE Generation Handler Due to External Kafka Issues #10
Conversation
Tanzim patch 1
|
|
|
||
|
|
||
|
|
||
| def mobility_data_generation(self) -> pd.DataFrame: |
There was a problem hiding this comment.
this method and generate_ue_data can be replaced with the original _mobility_data_generation, which can actually be moved to this directory (as you are correctly doing in this PR), into ue_tracks.py, and we may rename it as generate_as_lon_lat_points, and then fix the call-site.
There was a problem hiding this comment.
Updated the ue_tracks by adding _mobility_data_generation from services/ue_tracks_generation/u_tracks_generation_driver.py file to ue_tracks.py.
Renamed it to generate_as_lon_lat_points.
Will delete the reduntant ue_generation_handler.py after plot functions is fixed.
| return pd.concat(all_dataframes, ignore_index=True) | ||
|
|
||
|
|
||
| def plot_ue_tracks(self,csv_file) -> None: |
There was a problem hiding this comment.
make this a static method inside ue_tracks that takes a UETracksGenerator object and simulates and plots it. The whole simulation can be done and the whole data can be loaded into memory and plotted (as is being done here) OR, if possible, just enumerate through generate_as_lon_lat_points and plot one by one.
| # Update start_idx for the next batch | ||
| start_idx = end_idx | ||
|
|
||
| def save_data_to_csv(self, filename: str) -> None: |
There was a problem hiding this comment.
you may delete this (in future, you may provide such a utility if we need to build in more business logic around this export)
| ) | ||
|
|
||
|
|
||
| class UETracksGenerationParams: |
There was a problem hiding this comment.
Added the Unit Test Cases for the ue_tracks_params.
All of the changes are made.
There was a problem hiding this comment.
You check the second last commit to see the changes made in mobility/test module. The last commit was done to format all of the codes using Black.
* changed location for handler and added unit tests * fixed hanlder position * fixed import locations * used black on test & helper files * fixed name and removed redundant code in test * added docstring * added HOW TO RUN comment * blacked * fixed typo * made pr comment changes * added eu-tracks-params test code * fixed test_ue_params * added docstring * changed venv issues
|
Closing since done in #11 |
Overview
This PR introduces a local User Equipment (UE) generation handler to replace the current API service, which is failing due to unresolved Kafka connectivity issues. The new local handler ensures continued functionality of UE simulations without external dependencies.
Changes Made
Justification
The external API service for UE generation was reliant on Kafka streams, which are currently facing connectivity issues. To maintain productivity and ensure the reliability of our simulations, switching to a local generation method was deemed necessary.