I need help with my python code, I$\text{'}$m a beginner and I don't know what I$\text{'}$m doing wrong. The assignment instructions is in the screenshots. The test.py file runs my code which was written by the teacher. I can't fully add it here since there's not enough characters available. Is it also possible to compress my code without getting rid of the methods used? This is my code $($Assignment$2.$py$)$:

import numpy as np

class WeatherSimulation:

def $\_\_$init$\_\_($self$,$ transition$\_$probabilities, holding$\_$times$)$:

self.transition$\_$probabilities $=$ transition$\_$probabilities

self.holding$\_$times $=$ holding$\_$times

# Validate transition probabilities

for state, transitions in self.transition$\_$probabilities.items$()$:

if not np$.$isclose$($sum$($transitions$.$values$()),1\mathrm{.}0)$:

raise RuntimeError$($f$"$Transition probabilities for state $\text{'}\{$state$\}\text{'}$ do not sum to $1.")$

self.$\_$current$\_$state $=$ 'sunny' # Start with 'sunny' state

self.$\_$remaining$\_$hours $=$ holding$\_$times$[$self$.\_$current$\_$state$]$

def get$\_$states$($self$)$:

return list$($self$.$transition$\_$probabilities.keys$())$

def set$\_$state$($self$,$ state$)$:

# Set the weather state manually and reset remaining hours for that state

if state in self.transition$\_$probabilities:

self.$\_$current$\_$state $=$ state

self.$\_$remaining$\_$hours $=$ self.holding$\_$times$[$state$]$

else:

raise ValueError$($f$"$State $\text{'}\{$state$\}\text{'}$ is not a valid weather state."$)$

def current$\_$state$($self$)$:

# Return the current weather state

return self.$\_$current$\_$state

def current$\_$state$\_$remaining$\_$hours$($self$)$:

# Return the remaining hours for the current state

return self.$\_$remaining$\_$hours

def next$\_$state$($self$)$:

# Transition to the next weather state based on probabilities and reset remaining hours

next$\_$state $=$ np$.$random.choice$($

list$($self$.$transition$\_$probabilities$[$self$.\_$current$\_$state$].$keys$()),$

p$=$list$($self$.$transition$\_$probabilities$[$self$.\_$current$\_$state$].$values$())$

$)$

print$($f$"$Transitioning from $\{$self$.\_$current$\_$state$\}$ to $\{$next$\_$state$\}")$

self.$\_$current$\_$state $=$ next$\_$state

self.$\_$remaining$\_$hours $=$ self.holding$\_$times$[$next$\_$state$]$

def set$\_$holding$\_$time$\_$in$\_$state$($self$,$ state, hours$)$:

# Set holding time for a specific weather state

if state in self.holding$\_$times:

self.holding$\_$times$[$state$]=$ hours

else:

raise ValueError$($f$"$State $\text{'}\{$state$\}\text{'}$ does not exist."$)$

def iterable$($self$)$:

# This method returns an iterable generator that simulates the weather state transitions

while True:

yield self.$\_$current$\_$state, self.$\_$remaining$\_$hours

self.$\_$remaining$\_$hours $-=1$

print$($f$"$Iterating: State: $\{$self$.\_$current$\_$state$\},$ Remaining Hours: $\{$self$.\_$remaining$\_$hours$\}")$

if self.$\_$remaining$\_$hours :

self.next$\_$state$()$

def simulate$($self$,$ hours$)$:

occurrences $=\{$state: $0$ for state in self.get$\_$states$()\}$

total$\_$hours $=0$

while total$\_$hours $$ hours:

occurrences$[$self$.\_$current$\_$state$]+=1$

self.$\_$remaining$\_$hours $-=1$

total$\_$hours $+=1$

if self.$\_$remaining$\_$hours :

self.next$\_$state$()$

self.$\_$remaining$\_$hours $=$ self.holding$\_$times$[$self$.\_$current$\_$state$]$ # Reset after transition

total $=$ sum$($occurrences$.$values$())$

return $[$occurrences$[$state$]/$ total $*100$ for state in self.get$\_$states$()]$

# Example data for transition probabilities and holding times

my$\_$transitions $=\{$

'sunny': $\{\text{'}$sunny$\text{'}$: $0\mathrm{.}7,$ 'cloudy': $0\mathrm{.}3,$ 'rainy': $0,$ 'snowy': $0\},$

'cloudy': $\{\text{'}$sunny$\text{'}$: $0\mathrm{.}5,$ 'cloudy': $0\mathrm{.}3,$ 'rainy': $0\mathrm{.}15,$ 'snowy': $0\mathrm{.}05\},$

'rainy': $\{\text{'}$sunny$\text{'}$: $0\mathrm{.}6,$ 'cloudy': $0\mathrm{.}2,$ 'rainy': $0\mathrm{.}15,$ 'snowy': $0\mathrm{.}05\},$

'snowy': $\{\text{'}$sunny$\text{'}$: $0\mathrm{.}7,$ 'cloudy': $0\mathrm{.}1,$ 'rainy': $0\mathrm{.}05,$ 'snowy': $0\mathrm{.}15\}$

$\}$

my$\_$holding$\_$times $=\{\text{'}$sunny$\text{'}$: $1,$ 'cloudy': $2,$ 'rainy': $2,$ 'snowy': $1\}$

# Create the simulation object

simulation $=$ WeatherSimulation$($my$\_$transitions, my$\_$holding$\_$times$)$

# Run the simulation for $10\_000$ hours

result $=$ simulation.simulate$(10000)$

print$("$Percentage of time each state occurred: $",$ result$)$

END of my code. When running it through the test file it gets stuck on NOK! which means not ok:

if not check$\_$holding$\_$times$($transitions$,$ holding$\_$time$)$:

print$("$Probably a problem with holding times"$)$

sys$.$exit$(\text{'}$NOK$!\text{'})$

else:

print$("$Correct holding times"$)$